Commit 598a46e

committed

the key of set can be int

1 parent 61db5d1 commit 598a46eCopy full SHA for 598a46e

File tree

-7

lines changed

-7

lines changed

Lines changed: 6 additions & 7 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,6 @@`
`1`	`1`	`# Source : https://leetcode.com/problems/k-diff-pairs-in-an-array/?tab=Description`
`2`	`2`	`# Author : Han Zichi`
`3`		`-# Date : 2017-03-06`
	`3`	`+# Date : 2017-03-07`
`4`	`4`
`5`	`5`	`class Solution(object):`
`6`	`6`	`def findPairs(self, nums, k):`
`@@ -16,19 +16,18 @@ def findPairs(self, nums, k):`
`16`	`16`	`dict = {}`
`17`	`17`
`18`	`18`	`for item in nums:`
`19`		`- key = str(item)`
`20`		`- if key in dict:`
`21`		`- dict[key] += 1`
	`19`	`+ if item in dict:`
	`20`	`+ dict[item] += 1`
`22`	`21`	`else:`
`23`		`- dict[key] = 1`
	`22`	`+ dict[item] = 1`
`24`	`23`
`25`	`24`	`pre, ans = None, 0`
`26`	`25`	`for item in nums:`
`27`	`26`	`if item == pre:`
`28`	`27`	`continue`
`29`		`- dict[str(item)] -= 1`
	`28`	`+ dict[item] -= 1`
`30`	`29`	`target = item + k`
`31`		`- if str(target) in dict and dict[str(target)] > 0:`
	`30`	`+ if target in dict and dict[target] > 0:`
`32`	`31`	`ans += 1`
`33`	`32`	`pre = item`
`34`	`33`

Comments

(0)