K_pairs

题目说明

英文题目:
Given an array of integers and an integer k, you need to find the number of unique k-diff pairs in the array. Here a k-diff pair is defined as an integer pair (i, j), where i and j are both numbers in the array and their absolute difference is k.
中文题目(自己翻译):
给定一个整数k和整数组,在数组中找出所有的两个数相减的值为k的个数,要求(x,y)和(y,x)属于一个,只能计算一次。

例如1:

输入: [ 3,1,4,1,5 ],k = 2
输出: 2
说明:数组中有两对满足,即(1,3)和(3,5)。
虽然我们在输入中有两个1,但我们应该只返回唯一对的数量。

例如2:

输入: [ 1,2,3,4,5 ],k = 1
输出: 4
说明:在阵列中有四对,(1,2),(2,3),(3,4)和(4,5)。

例如3:

输入: [ 1,3,1,5,4 ],k = 0
输出: 1
说明:数组中有一对,(1,1)。

解题思路

首先,需要注意:

(1)(x,y)和(y,x)是相同的;  
(2)k为负数的时候,返回0;k为正数的时候分情况;

当k为正数的时候,又分为k=0和k!=0 的情况。
k != 0的情况:

(1)排序;
(2)去重;
(3)数组内两两相减,判断是否等于k,相等则计数器+1;

k = 0的情况:

(1)排序;
(2)去重,去重的时候需要注意,不是讲所有的重复的值去掉,是将两个或两个以上的重复的值去掉之后变成只有两个重复的值;
(3)数组内两两相减,判断是否等于k,相等则计数器+1;

这两种情况下的去重的格式是不一样的。

代码

有了上述的解题思路之后,需要考虑到时间复杂度和空间复杂度的。

方法一,时间复杂度O(n^2):

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var findPairs = function(nums, k) {
    
    // 判断k是否小于0或者数组的长度是否为0,是则返回0。
    if(nums.length === 0 || k<0) return 0;
    
    // 将数组从小到大排序
     nums.sort((x,y) => {return x-y;})
    
     // 两种情况下的去重
     if(k !== 0) {
         let setArr = new Set(nums);
         nums = [...setArr];
         // console.log(nums)
     }else{
         for(let i = 2; i<nums.length;i++ ){
             if(nums[i-2] === nums[i]) {
                 nums.splice(i,1);
                 i--;
             }
         }
     }
    
     let length = nums.length;
     let count = 0;

    // 数组内两两相减,判断是否等于k,相等则计数器count+1
     for(let i = 0; i<length-1; i++) {
         for(let j = i+1 ;j<length ;j++) {
             let D_value = (nums[j] - nums[i]);
             if(D_value === k) {
                 count++;
                 break;
             }
         }
     }
    
     return count;
};

方法二,时间复杂度O(n):

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var findPairs = function(nums, k) {
    
    // 判断k是否小于0或者数组的长度是否为0,是则返回0。
    if(nums.length === 0 || k<0) return 0;

    if(k !== 0) {
        // 去重,记录下去重之后的长度。
        let setArr = new Set(nums);
        let size = setArr.size;
        nums = [...setArr];
        
        // 转化成数组之后加上k,并将其添加到setArr里边;
        for(let i = 0 ;i<size;i++) {
            // nums[i] += k;
            setArr.add(nums[i]+k);
        }
        
        // 返回满足题目的对数,size*2表示没有重复时setArr的长度,减去setArr的长度,就是重复的个数,也就是我们要的值。
        return (size*2 - setArr.size);
    }else{
        // 排序
        nums.sort((x,y) => {return x-y;});
        
        // 将两个或两个以上的重复的值变成两个
        for(let i = 2; i<nums.length;i++ ){
            if(nums[i-2] === nums[i]) {
                nums.splice(i,1);
                i--;
            }
        }
        
        // 去重之后计算长度,用数组长度减去去重之后的长度,就表示k个重复的值。
        let size = (new Set(nums)).size;
        
        return nums.length-size;
    }
};

结论

本题主要需要考虑到k大于0 和等于0的情况,因为在两种情况下所使用的去重方法不同;

其次,就是在等于0 的时候所用的去重方法,不是将所有的都去掉,而是要将两个以上的保留成两个。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,928评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,192评论 3 387
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,468评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,186评论 1 286
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,295评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,374评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,403评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,186评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,610评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,906评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,075评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,755评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,393评论 3 320
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,079评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,313评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,934评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,963评论 2 351

推荐阅读更多精彩内容