【HackerRank】Pairs

题目链接：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[n];

         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);

     }

 }

巴特西

【HackerRank】Pairs

最新文章

热门文章