Commit 9e799bf

author

Shuo

authored

Merge pull request #308 from openset/develop

Add: Sum of Even Numbers After Queries

2 parents ad68398 + 73eef3d commit 9e799bfCopy full SHA for 9e799bf

File tree

+50

-0

lines changed

+50

-0

lines changed

Lines changed: 22 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,22 @@`
	`1`	`+package sum_of_even_numbers_after_queries`
	`2`	`+`
	`3`	`+func sumEvenAfterQueries(A []int, queries [][]int) []int {`
	`4`	`+ sum, ans := 0, make([]int, 0)`
	`5`	`+ for _, v := range A {`
	`6`	`+ if v%2 == 0 {`
	`7`	`+ sum += v`
	`8`	`+ }`
	`9`	`+ }`
	`10`	`+ for _, q := range queries {`
	`11`	`+ v, i := q[0], q[1]`
	`12`	`+ if A[i]%2 == 0 {`
	`13`	`+ sum -= A[i]`
	`14`	`+ }`
	`15`	`+ A[i] += v`
	`16`	`+ if A[i]%2 == 0 {`
	`17`	`+ sum += A[i]`
	`18`	`+ }`
	`19`	`+ ans = append(ans, sum)`
	`20`	`+ }`
	`21`	`+ return ans`
	`22`	`+}`

Lines changed: 28 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,28 @@`
	`1`	`+package sum_of_even_numbers_after_queries`
	`2`	`+`
	`3`	`+import (`
	`4`	`+ "reflect"`
	`5`	`+ "testing"`
	`6`	`+)`
	`7`	`+`
	`8`	`+type caseType struct {`
	`9`	`+ input []int`
	`10`	`+ queries [][]int`
	`11`	`+ expected []int`
	`12`	`+}`
	`13`	`+`
	`14`	`+func TestSumEvenAfterQueries(t *testing.T) {`
	`15`	`+ tests := [...]caseType{`
	`16`	`+ {`
	`17`	`+ input: []int{1, 2, 3, 4},`
	`18`	`+ queries: [][]int{{1, 0}, {-3, 1}, {-4, 0}, {2, 3}},`
	`19`	`+ expected: []int{8, 6, 2, 4},`
	`20`	`+ },`
	`21`	`+ }`
	`22`	`+ for _, tc := range tests {`
	`23`	`+ output := sumEvenAfterQueries(tc.input, tc.queries)`
	`24`	`+ if !reflect.DeepEqual(output, tc.expected) {`
	`25`	`+ t.Fatalf("input: %v, output: %v, expected: %v", tc.input, output, tc.expected)`
	`26`	`+ }`
	`27`	`+ }`
	`28`	`+}`

Comments

(0)