the key of set can be int

Author: lessfish

Algorithms/K-diff Pairs in an Array/k-diff-pairs-in-an-array.py

@@ -1,6 +1,6 @@
 # Source : https://leetcode.com/problems/k-diff-pairs-in-an-array/?tab=Description
 # Author : Han Zichi
-# Date   : 2017-03-06
+# Date   : 2017-03-07
 
 class Solution(object):
     def findPairs(self, nums, k):
@@ -16,19 +16,18 @@ def findPairs(self, nums, k):
         dict = {}
 
         for item in nums:
-            key = str(item)
-            if key in dict:
-                dict[key] += 1
+            if item in dict:
+                dict[item] += 1
             else:
-                dict[key] = 1
+                dict[item] = 1
 
         pre, ans = None, 0
         for item in nums:
             if item == pre:
                 continue
-            dict[str(item)] -= 1
+            dict[item] -= 1
             target = item + k
-            if str(target) in dict and dict[str(target)] > 0:
+            if target in dict and dict[target] > 0:
                 ans += 1
             pre = item