ท้าทาย

รับเมทริกซ์Mพร้อมแถวrและคอลัมน์cและ Boolean สองรายการแสดงรายการVของความยาวrและHของความยาวcให้คำนวณผลรวมแนวตั้งและแนวนอนที่แบ่งพาร์ติชันแล้ว

กฎระเบียบ

• rและcมากกว่าหรือเท่ากับหนึ่ง

• HและVเริ่มต้นด้วยมูลค่าที่แท้จริง

• ค่าในMอยู่ภายในโดเมนตัวเลขที่สมเหตุสมผลของภาษาของคุณ

• การแบ่งพาร์ติชันและการสรุปเริ่มต้นที่มุมซ้ายบน

เดินผ่าน

ให้M :

┌──────────────┐
│ 1  2  3  4  5│
│ 6  7  8  9 10│
│11 12 13 14 15│
│16 17 18 19 20│
└──────────────┘

H :1 0 1 0 0

V :1 1 0 1

แยกMเป็นกลุ่มของคอลัมน์เริ่มต้นกลุ่มใหม่ในทุกมูลค่าที่แท้จริงของH

┌─────┬────────┐
│ 1  2│ 3  4  5│
│ 6  7│ 8  9 10│
│11 12│13 14 15│
│16 17│18 19 20│
└─────┴────────┘

แยกกลุ่มคอลัมน์แต่ละกลุ่มออกเป็นกลุ่มแถวเริ่มกลุ่มใหม่ด้วยค่าจริงทุกV :

┌─────┬────────┐
│ 1  2│ 3  4  5│
├─────┼────────┤
│ 6  7│ 8  9 10│
│11 12│13 14 15│
├─────┼────────┤
│16 17│18 19 20│
└─────┴────────┘

สรุปผลรวมของแต่ละเซลล์ในแนวนอน:

┌─────┬────────┐
│ 1  3│ 3  7 12│
├─────┼────────┤
│ 6 13│ 8 17 27│
│11 23│13 27 42│
├─────┼────────┤
│16 33│18 37 57│
└─────┴────────┘

สรุปผลรวมของแต่ละเซลล์ในแนวตั้ง:

┌─────┬────────┐
│ 1  3│ 3  7 12│
├─────┼────────┤
│ 6 13│ 8 17 27│
│17 36│21 44 69│
├─────┼────────┤
│16 33│18 37 57│
└─────┴────────┘

ผลลัพธ์:

┌──────────────┐
│ 1  3  3  7 12│
│ 6 13  8 17 27│
│17 36 21 44 69│
│16 33 18 37 57│
└──────────────┘

กรณีทดสอบเพิ่มเติม

ม :

┌───────────┐
│15 11 11 17│
│13 20 18  8│
└───────────┘

H : 1 0 0 1 V :1 0

ผลลัพธ์:

┌───────────┐
│15 26 37 17│
│28 59 88 25│
└───────────┘

ม :

┌─┐
│7│
└─┘

ผลลัพธ์ ( HและVต้องเป็น1):

┌─┐
│7│
└─┘

ม :

┌──┐
│ 3│
│-1│
│ 4│
└──┘

V : 1 1 0( ต้องเป็นH1 )

ผลลัพธ์:

┌──┐
│ 3│
│-1│
│ 3│
└──┘

ม :

┌───────────────────────────────────────────────────────┐
│10    7.7 1.9 1.5 5.4  1.2 7.8 0.6 4.3 1.2  4.5 5.4 0.3│
│ 2.3  3.8 4.1 4.5 1    7.7 3   3.4 6.9 5.8  9.5 1.3 7.5│
│ 9.1  3.7 7.2 9.8 3.9 10   7.6 9.6 7.3 6.2  3.3 9.2 9.4│
│ 4.3  4.9 7.6 2   1.4  5.8 8.1 2.4 1.1 2.3  7.3 3.6 6  │
│ 9.3 10   5.8 9.6 5.7  8.1 2.1 3.9 4   1.3  6.3 3.1 9  │
│ 6.6  1.4 0.5 6.5 4.6  2.1 7.5 4.3 9   7.2  2.8 3.6 4.6│
│ 1.7  9.9 2.4 4.5 1.3  2.6 6.4 7.8 6.2 3.2 10   5.2 8.9│
│ 9.9  5.3 4.5 6.3 1.4  3.1 2.3 7.9 7.8 7.9  9.6 4   5.8│
└───────────────────────────────────────────────────────┘

H :1 0 0 1 0 1 1 1 0 1 1 1 0

V :1 0 0 0 0 1 0 0

ผลลัพธ์:

┌────────────────────────────────────────────────────────────────┐
│10   17.7 19.6  1.5  6.9  1.2  7.8  0.6  4.9  1.2  4.5  5.4  5.7│
│12.3 23.8 29.8  6   12.4  8.9 10.8  4   15.2  7   14    6.7 14.5│
│21.4 36.6 49.8 15.8 26.1 18.9 18.4 13.6 32.1 13.2 17.3 15.9 33.1│
│25.7 45.8 66.6 17.8 29.5 24.7 26.5 16   35.6 15.5 24.6 19.5 42.7│
│35   65.1 91.7 27.4 44.8 32.8 28.6 19.9 43.5 16.8 30.9 22.6 54.8│
│ 6.6  8    8.5  6.5 11.1  2.1  7.5  4.3 13.3  7.2  2.8  3.6  8.2│
│ 8.3 19.6 22.5 11   16.9  4.7 13.9 12.1 27.3 10.4 12.8  8.8 22.3│
│18.2 34.8 42.2 17.3 24.6  7.8 16.2 20   43   18.3 22.4 12.8 32.1│
└────────────────────────────────────────────────────────────────┘

เยลลี่ 10 ไบต์

Zœṗ@+\€Ẏð/

ลองออนไลน์! และกรณีทดสอบครั้งสุดท้าย (ด้วยGเมื่อสิ้นสุดการอ่าน)

อินพุตถูกใช้เป็นรายการ [M, H, V]การป้อนข้อมูลจะนำมาเป็นรายการ

คำอธิบาย

Zœṗ@+\€Ẏð/  Input: [M, H, V]
        ð/  Insert the previous (f) as a dyadic link
            Forms f( f(M, H) , V)
            For f(x, y):
Z             Transpose x
 œṗ@          Partition the rows of x^T at each true in y
    +\€       Compute the cumulative sums in each partition
       Ẏ      Tighten (Joins all the lists at the next depth)

APL (Dyalog) 13 ไบต์

รับ ist ของVHMเป็นอาร์กิวเมนต์

{⍉⊃,/+\¨⍺⊂⍵}/

ลองออนไลน์!

{… }/ insert (ลดโดย) ฟังก์ชั่นที่ไม่ระบุชื่อต่อไปนี้ซึ่งคำทางด้านซ้ายแสดงด้วย⍺และคำทางด้านขวาแสดงด้วย⍵ เนื่องจากฟังก์ชั่น APL มีความสัมพันธ์ที่ถูกต้องดังนั้นนี่คือV f ( H f M )

⍺⊂⍵ พาร์ติชัน⍵ตาม⍺

+\¨ ผลรวมสะสมของแต่ละส่วน

,/ ลดการเรียงต่อกัน (สิ่งนี้ล้อมรอบผลลัพธ์เพื่อลดอันดับ)

⊃ เปิดเผย

⍉ transpose

Python 2 + จำนวน 143 138 117 115 110 108 ไบต์

^{-21 ไบต์ขอบคุณAdám !}

lambda M,*L:reduce(lambda m,l:vstack(map(lambda p:cumsum(p,0),split(m,*where(l)))).T,L,M)
from numpy import*

ลองออนไลน์!

เยลลี่ ,  15  14 ไบต์

œṗ+\€Ẏ
ḢçЀZð⁺

การเชื่อมโยง dyadic H,Vที่ด้านซ้ายและMด้านขวาและกลับเมทริกซ์ที่เกิด

ลองออนไลน์!

อีกวิธีหนึ่งเป็นบรรทัดเดียวสำหรับ 14: Ḣœṗ+\€Ẏ$¥Ð€Zð⁺

อย่างไร?

œṗ+\€Ẏ - Link 1: partition and cumSum: list of partition bools, list of values
œṗ     - partition (the values) at truthy indexes (of the bools)
    €  - for €ach part:
  +\   -   cumulative reduce by addition
     Ẏ - tighten (flattens back into a list)

ḢçЀZð⁺ - Main link: list of lists, [H,V]; list of lists, M
      ⁺ - perform this twice:
     ð  - [it's a dyadic chain for the second pass, first pass is dyadic implicitly]
Ḣ       -   head, pop it & modify (so H the first time, V the second)
  Ѐ    -   map across right: (M the first time, the intermediate result the second)
 ç      -     the last link (1) as a dyad
    Z   -   transpose the result (do the rows first time, and the columns the second)

ก่อนหน้านี้:

œṗ@+\€Ẏ
ç€Zç€⁵Z

โปรแกรมเต็มรูปแบบที่พิมพ์การแสดงผลลัพธ์

MATL , 19 ไบต์

,!ix"0GYs@12XQ!g]v!

อินพุตคือM(เมทริกซ์), H(เวกเตอร์คอลัมน์)V (เวกเตอร์คอลัมน์) ตัวคั่นแถวคือ;คั่นแถว

ลองออนไลน์! หรือตรวจสอบกรณีทดสอบทั้งหมด: 1 , 2 , 3 , 4 , 5 5

คำอธิบาย

นี่คือผลรวมสะสมในแนวนอนจากนั้นในแนวตั้ง

,          % Do the following twice
  !        %   First time this inputs M implicitly. Transpose. Second time
           %   it transposes the result of the horizontal cumulative sum
  ix       %   Input H (first time) or V (second time). Delete it; but gets
           %   copied into clipboard G
  "        %   For each column of the matrix
    0G     %     Push most recent input: H (first time) or V (second)
    Ys     %     Cumulative sum. This produces a vector of integer values
           %     such that all columns (first time) or rows (second) of M 
           %     with the same value in this vector should be cumulatively
           %     summed
    @      %     Push current column of M transposed (first time) or M after
           %     horizontal cumulative sum (second time)
    12XQ   %     Cumulative sum. Gives a cell array of row vectors
    !g     %     Join those vectors into one row vector
  ]        %   End
  v        %   Concatenate the row vectors vertically into a matrix
  !        %   Transpose. This corrects for the fact that each column vector
           %   of the matrix was cumulatively summed into a row vector
           % Implicit end. Implicit display

J , 20 ไบต์

;@(<@(+/\);.1|:)&.>/

ลองออนไลน์!

อินพุตถูกใช้เป็นอาร์เรย์ของกล่องที่มี [V, H, M]การป้อนข้อมูลจะมาเป็นอาร์เรย์ของกล่องที่มี

คำอธิบาย

;@(<@(+/\);.1|:)&.>/  Input: [V H M]
  (     g      )   /  Insert g and reduce (right-to-left)
                      Forms V g H g M = V g (H g M)
                & >     Unbox each
             |:         Transpose the right arg
          ;.1           Partition
      +/\               Reduce each prefix using addition (cumulative sum)
   <@                   Box each partition
;@                      Raze (Concatenate the contents in each box)
                &.>     Box the result

Mathematica, 212 ไบต์

(T=Transpose;A=AppendTo;J=Flatten;f[s_]:=Block[{},t=2;r=1;w={};While[t<=Length@s,If[s[[t]]==0,r++,w~A~r;r=1];t++];w~A~r];K[x_,y_]:=Accumulate/@#&/@(FoldPairList[TakeDrop,#,f@y]&/@x);d=J/@K[#,#2];T[J/@K[T@d,#3]])&

อินพุต
[M, H, V]

[{{15, 11, 11, 17}, {13, 20, 18, 8}}, {1, 0, 0, 1}, {1, 0}]

C # (. NET Core) 164 ไบต์

(M,H,V)=>{int a=M.Length,b=M[0].Length,i,j;for(i=0;i<a;i++)for(j=0;j<b;j++)if(!H[j])M[i][j]+=M[i][j-1];for(i=0;i<a;i++)for(j=0;j<b;j++)if(!V[i])M[i][j]+=M[i-1][j];}

ลองออนไลน์!

โดยพื้นฐานแล้วมันจะตรงตามที่ระบุใน OP มันจะวนซ้ำในแนวนอนก่อนแล้วจึงวนซ้ำตามแนวตั้ง

Haskell , 129 ไบต์ 119 ไบต์

s m v=tail$scanl(\a(x,s)->if s then x else zipWith(+)a x)[](zip m v)
t=foldr(zipWith(:))$repeat[]
f m h v=t$s(t$s m v)h

ลองออนไลน์!

บันทึกแล้ว 10 ไบต์ขอบคุณ @ceasedtoturncounterclockwis

t(สำหรับการย้าย) สลับแถวและคอลัมน์ คำอธิบายอย่างรวดเร็ว:

foldr(zipWith(:))(repeat[])(r1,...,rn) =
zipWith(:) r1 (zipWith(:) r2 (... zipWith(:) rn (repeat [])))

อ่านจากขวาไปซ้าย: เราเรียกดูแถวจากล่างขึ้นบนและกดแต่ละค่าในคอลัมน์ปลายทาง

s นั้นเป็นผลรวมแบบหมุนของเวกเตอร์ แต่จะรีเซ็ตเมื่อค่า True เกิดขึ้น v

fผลรวมของแถวด้วยการsติดตามvและทำเช่นเดียวกันกับคอลัมน์ต่อไปนี้h

JavaScript (ES6), 88 ไบต์

(a,h,v)=>a.map(b=>b.map((e,i)=>t=h[i]?e:t+e)).map((b,j)=>t=v[j]?b:t.map((e,i)=>e+b[i]))

เยลลี่ 31 ไบต์

+\€€
œṗḊZ€⁵œṗ$€Ḋ€Ç€ÇZ€€Z€;/€€;/

ลองออนไลน์!

นี่เป็นเวลานานเกินไปสำหรับ Jelly xD

BTW, 11/31 ไบต์ในโปรแกรมนี้ประกอบด้วยอักขระยูโร นั่นเป็นหนึ่งในสามของโปรแกรม!