数组中的K-diff数对

旧城等待， 2022-05-10 01:54 135阅读 0赞

> 给定一个整数数组和一个整数 **k**, 你需要在数组里找到**不同的 **k-diff 数对。这里将 **k-diff** 数对定义为一个整数对 (i, j), 其中** i **和** j **都是数组中的数字，且两数之差的绝对值是 **k**.

**示例 1:**

>     输入: [3, 1, 4, 1, 5], k = 2
>     输出: 2
>     解释: 数组中有两个 2-diff 数对, (1, 3) 和 (3, 5)。
>     尽管数组中有两个1，但我们只应返回不同的数对的数量。

**示例 2:**

>     输入:[1, 2, 3, 4, 5], k = 1
>     输出: 4
>     解释: 数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5)。

**示例 3:**

>     输入: [1, 3, 1, 5, 4], k = 0
>     输出: 1
>     解释: 数组中只有一个 0-diff 数对，(1, 1)。

**注意:**

1.  数对 (i, j) 和数对 (j, i) 被算作同一数对。
2.  数组的长度不超过10,000。
3.  所有输入的整数的范围在 \[-1e7, 1e7\]。

**代码实现：**

class Solution {
    public:
      int findPairs(vector<int>& nums, int k) {
          if (k < 0) return 0;
          unordered_map<int, int> data;
          
    /*
          for(int i = 0;i<nums.size();i++) data[nums[i]]++;//等价下式
    */
    
          for (int n : nums) data[n]++;//初始化，键为nums[n]，值为data[n]++,初始都为0，出现次数
          //    3 1
          //    1 2
          //    4 1
          //    5 1
          int cnt = 0;
          for (auto temp : data) {//自动的类型，此处temp类型为unordered_map<int, int>
              if ((!k && temp.second > 1)      //当k=0时候，只需要知道表中有次数大于1的就可成立
                  || (k && data.count(temp.first + k))) ++cnt;  //K大于1时，在data中查找，temp.first+k的值即可
          }
          return cnt;        
      }
    };