整数对查找

题目

请设计一个高效算法，找出数组中两数之和为指定值的所有整数对。
给定一个int数组A和数组大小n以及需查找的和sum，请返回和为sum的整数对的个数。保证数组大小小于等于3000。

测试样例：
[1,2,3,4,5],5,6
返回：2

解答

1.原始暴力解法：复杂度比较高

import java.util.*;

public class FindPair {
    public int countPairs(int[] A, int n, int sum) {
        // write code here
        int count = 0;
        Arrays.sort(A);
        for(int i = 0;i < n;i++){
            for(int j = i + 1;j < n;j++){
                if(A[j] == sum - A[i] ){
                    count++;
                }
            }
        }
        return count;
    }
}

2.使用哈希表
使用哈希表存储对应数字的个数，然后通过getOrDefault(sum-temp,0)方法获得对应数字的个数，最后算出结果。
put方法
V.put(K key, V value)

getOrDefault()方法
V.getOrDefaultt(K key, D defualt) —> return value of key.
当Map集合中有这个key时，就使用这个key值，如果没有就使用默认值defaultValue

import java.util.*;
 
public class FindPair {
    public int countPairs(int[] A, int n, int sum) {
        HashMap<Integer,Integer> map=new HashMap<>();
        int count=0;
        for(int temp:A){
            count+=map.getOrDefault(sum-temp,0);//获得k值
            map.put(temp,map.getOrDefault(temp,0)+1);//保存个数
        }
        return count;
    }
}