您的位置:首页 > 大数据 > 人工智能

【HackerRank】Pairs

2014-08-15 15:52 330 查看
题目链接:Pairs

完全就是Two Sum问题的变形!Two Sum问题是要求数组中和正好等于K的两个数,这个是求数组中两个数的差正好等于K的两个数。总结其实就是“骑驴找马”的问题:即当前遍历ar[i],那么只要看数组中是否存在ar[i]+K或者ar[i]-K就可以了,还是用HashMap在O(1)的时间完成这个操作。

题目有一点没说清楚的就是元素是否有重复,从Editorial来看似乎是没有重复,不过我还是用map的value记录了数出现的频率来处理了重复。

代码如下:

import java.util.*;

public class Solution {
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = in.nextInt();
int k = in.nextInt();
HashMap<Integer, Integer> map = new HashMap<Integer,Integer>();
int[] ar = new int
;
for(int i = 0;i < n;i ++){
ar[i] = in.nextInt();
if(!map.containsKey(ar[i]))
map.put(ar[i], 0);
map.put(ar[i], map.get(ar[i])+1);
}

int answer = 0;
for(int i = 0;i < n;i ++){
int up = ar[i]+k;
if(map.containsKey(up))
answer += map.get(up);
int down = ar[i]-k;
if(map.containsKey(down))
answer += map.get(down);
}
System.out.println(answer/2);

}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: