【算法练习】日常刷题记录

一、四数相加Ⅱ（中等）HashMap、脑筋急转弯

把A+B+C+D，拆分成A+B和C+D，先用两个嵌套循环遍历AB，用Map存储A+B对应和出现的次数，再用两个嵌套循环遍历CD，在之前的Map中找**-(C+D)，这样一加就=0，每次找到了就加上Map中存储的A+B出现的次数。**

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        HashMap<Integer, Integer> one = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            for (int j = 0; j < nums2.length; j++) {
                one.put(nums1[i] + nums2[j], one.getOrDefault(nums1[i] + nums2[j], 0) + 1);
            }
        }
        int ans = 0;
        for (int i = 0; i < nums3.length; i++) {
            for (int j = 0; j < nums4.length; j++) {
                if (one.containsKey(-(nums3[i] + nums4[j]))) {
                    ans += one.get(-(nums3[i] + nums4[j]));
                }
            }
        }
        return ans;
    }
}

二、三数之和（中等）双指针

三个数组，三重循环impossible，又是要求目标和为0，可以考虑双指针（三指针），可以先把第一个元素确定下来，再去确定一个元素，然后再遍历第三个元素，为了避免重复结果，需要进行排序，然后使用之前多次使用的去重方法，可以多加一些限制条件以加快代码执行。排序后，如果第一个数大于0，就可以直接break，不用考虑后续数字。我们让第三个数字指在末尾位置，再开始遍历第二个数字，第二个数字从第一个数字后面开始遍历，如果三个数字之和大于0，首先从第三个数开始移动（因为已经排过序），在遍历下一个第二个数字时，第三个数字的下标就用之前的，不要重置，因为数组排过序，下一个第二个数字肯定大于上一个，第三个数字只能越来越往左边靠（每次遍历第二个数字时重置第三个数字也可以，只是时间开销很大，和使用三重循环没什么区别，使用三指针+排序就是为了把n^3 降到 n^2）。

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> ans = new ArrayList<>();
        for (int i = 0; i < nums.length - 2; i++) {
            if (nums[i] > 0) {
                break;
            }
            // 去重
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }

            int k = nums.length - 1;
            // 第三个数字从数组末尾开始遍历
            for (int j = i + 1; j < nums.length - 1; j++) {
                List<Integer> tmp = new ArrayList<>();
                // 继续去重
                if (j > i + 1 && nums[j] == nums[j - 1]) {
                    continue;
                }
                // -2 -1 1 2
                // 如果当前三数之和 大于 0，移动第三个数k
                while (j < k && nums[i] + nums[j] + nums[k] > 0) {
                    k--;
                }
                if (j >= k) {
                    break;
                }
                // 满足题目要求
                if (nums[i] + nums[j] + nums[k] == 0) {
                    tmp.add(nums[i]);
                    tmp.add(nums[j]);
                    tmp.add(nums[k]);
                    ans.add(new ArrayList<>(tmp));
                }
            }
        }
        return ans;
    }
}

三、四数之和（中等）双指针（同上面题）

就是多加了一层遍历，注意最后一个元素的声明位置！

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        // 同三数之和，用双指针加快搜索速度
        Arrays.sort(nums);
        List<List<Integer>> ans = new ArrayList<>();
        for (int i = 0; i < nums.length; i++) {
            // 去重
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            for (int j = i + 1; j < nums.length; j++) {
                // 去重
                if (j > i + 1 && nums[j] == nums[j - 1]) {
                    continue;
                }
                // 注意第四个数的声明位置，位置是双指针的核心
                int m = nums.length - 1;
                for (int k = j + 1; k < nums.length; k++) {
                    List<Integer> tmp = new ArrayList<>();
                    // 去重
                    if (k > j + 1 && nums[k] == nums[k - 1]) {
                        continue;
                    }
                    // 因为排过序，大于零优先移动最右边的数
                    while (k < m && nums[i] + nums[j] + nums[k] + nums[m] > target) {
                        m--;
                    }
                    // 每个元素只能用一次
                    if (k == m) {
                        break;
                    }
                    if (nums[i] + nums[j] + nums[k] + nums[m] == target) {
                        tmp.add(nums[i]);
                        tmp.add(nums[j]);
                        tmp.add(nums[k]);
                        tmp.add(nums[m]);
                        ans.add(new ArrayList<>(tmp));
                    }
                }
            }
        }
        return ans;
    }
}