Commit 6f1038b

committed

Solve 974. Subarray Sums Divisible by K

1 parent 33574ad commit 6f1038bCopy full SHA for 6f1038b

File tree

+28

-0

lines changed

+28

-0

lines changed

Lines changed: 28 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,28 @@`
	`1`	`+class Solution {`
	`2`	`+`
	`3`	`+ fun subarraysDivByK(nums: IntArray, k: Int): Int {`
	`4`	`+ var numberOfSubarrays = 0`
	`5`	`+ var sum = 0`
	`6`	`+ val remainders = IntArray(k) { 0 }`
	`7`	`+`
	`8`	`+ nums.forEach { num ->`
	`9`	`+ sum += num`
	`10`	`+ var remainder = sum % k`
	`11`	`+`
	`12`	`+ if (remainder < 0) {`
	`13`	`+ remainder = (remainder + k) % k`
	`14`	`+ }`
	`15`	`+`
	`16`	`+ remainders[remainder] += 1;`
	`17`	`+ }`
	`18`	`+`
	`19`	`+ (0 until k).forEach { i ->`
	`20`	`+ if (remainders[i] > 1) {`
	`21`	`+ numberOfSubarrays += (remainders[i] * (remainders[i] - 1)) / 2;`
	`22`	`+ }`
	`23`	`+ }`
	`24`	`+`
	`25`	`+ return remainders[0] + numberOfSubarrays`
	`26`	`+ }`
	`27`	`+`
	`28`	`+}`

Comments

(0)