Leetcode 532. K-diff Pairs in an Array

来源:互联网 发布:剑三正太脸型数据网盘 编辑:程序博客网 时间:2024/06/05 19:46

题目要求找出数组中差值为k的对数,要求有重复元素出现时不可重复计数。



两种思路求解:


1. 先将数组排序,再设定指针定位元素位置。NOTE:使用while循环排除掉重复元素。

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
      
        if(k<0) return 0;
        int res = 0;
        sort(nums.begin(), nums.end());
        
        for(int i=0, j=1, n=nums.size(); i<n && j<n;)
        {
            if(i==j || nums[j]-nums[i]<k) j++;
            else if(nums[j]-nums[i]>k) i++;
            else{
                res++;
                while(j<n && nums[j]-nums[i]==k) j++;
                while(j<n-1 && nums[j]==nums[j+1]) j++;
                int ii = i;
                if(i<j && nums[ii]==nums[i]) i++;
                
            }
        }
        
        return res;
    }
};


2. 使用map记录元素键值,通过判断n+k、n-k是否在map中出现过改变返回元素res的值(即差为k的数目)。

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
 
        if(k<0) return 0;
        int res = 0;
        unordered_map<int, int> my;
        for(int &n:nums)
        {
            if(my.count(n-k) && !my[n-k]) ++res, ++my[n-k];
            if(my.count(n+k) && !my[n]) ++res, ++my[n];
            if(!my.count(n)) my[n]=0;
        }
        return res;
        
    }
};


最近做题有点拖延,遇到不会的点立马去找不要等。方法一的指针定位元素位置的题做过好几次了,仍然不能熟练运用。中间停顿的时间太长,题感会变差。每天做几道,逐渐找回状态。