Commit 6dbb29e

authored

Create Subarray_Sum_Equals_K.py

1 parent 6f4e508 commit 6dbb29eCopy full SHA for 6dbb29e

File tree

+20

-0

lines changed

+20

-0

lines changed

Lines changed: 20 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,20 @@`
	`1`	`+class Solution:`
	`2`	`+ def subarraySum(self, nums: List[int], k: int) -> int:`
	`3`	`+ ans, pre = 0, 0`
	`4`	`+ d = {}`
	`5`	`+ for i in nums:`
	`6`	`+ pre += i`
	`7`	`+`
	`8`	`+ if pre == k:`
	`9`	`+ ans += 1`
	`10`	`+`
	`11`	`+ if pre - k in d:`
	`12`	`+ ans += d[pre-k]`
	`13`	`+`
	`14`	`+ if pre not in d:`
	`15`	`+ d[pre] = 1`
	`16`	`+ else:`
	`17`	`+ d[pre] += 1`
	`18`	`+`
	`19`	`+`
	`20`	`+ return ans`

Comments

(0)