【面试笔试真题】Java实现小于M的两数组合

用例输入

输入{a1, a2, a3…an}，再输入一个值M

用例输出

输出数列A中小于M的两数组合

示例

输入：
7 -1 -1
9

输出：
3

输出说明：
（7，-1）、（7，-1）、（-1，-1）总共3组，同一数据只能使用一次，相同值的不同数据可以一起使用。

解题思路

首先解决一下输入的问题，因为没有设定数组A的长度，所以只能用scanner.hasNext()来读取数据，将数据存入ArrayList中，ArrayList中最后一个数据就是M，需要提前读取并从ArrayList中删除。
之后对于数据的处理有两种思路：
一是暴力求解，使用嵌套for循环遍历ArrayList，时间复杂度为O(n^2)，但这肯定不是最佳解法，最后使用暴力解法的用例通过率为40%。
二是双指针差值求解，因为仅考虑两数之和，在有序数组中，设置头尾指针，若头指针start与最小的尾指针end小于M，则start与(start,end]这个区间的全部组合都小于M。时间复杂度为O(n)，最后用例通过100%。

代码实现

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        HashSet<Integer> set = new HashSet<>();
        ArrayList<Integer> list = new ArrayList<>();
        while(sc.hasNext()){
            list.add(sc.nextInt());
        }
        //最后一个为M
        int sum = list.get(list.size()-1);
        list.remove(list.size()-1);
        Collections.sort(list);
        int result = 0;
        for(int start = 0, end = list.size()-1; start < end;){
            if(list.get(start) + list.get(end) <= sum){
                result += end - start;
                start++;
            }else{
                end--;
            }
        }
        System.out.println(result);
    }
}