代码随想录打卡Day6

Leetcode 454

题目链接:四数相加 II
题目分了四个数组,每个里面取一个数,不需要考虑去重的问题,相对简单,将四个数组两两分组:

  1. 把前两个数组求和的每一种可能都加入HashMap中(key是和,value是出现的个数)
  2. 对后两个数组每一次求和后,检查HashMap中是否存在和为0的数,若有则结果加上该数次数
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int n = nums1.length;
        int result=0;
        HashMap<Integer,Integer> hashmap = new HashMap<>();
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){
                hashmap.put(-nums1[i]-nums2[j], hashmap.getOrDefault(-nums1[i]-nums2[j], 0)+1);
            }
        }
        for(int i=0; i<n; i++){
            for(int j=0; j<n; j++){
                if(hashmap.containsKey(nums3[i] + nums4[j])){
                    result += hashmap.getOrDefault(nums3[i] + nums4[j], 0);
                }
            }
        }

        return result;
    }

LeetCode 15

题目链接:三数之和
一个数组里面找出所有不重复的三个元素,满足和为target
采用双指针法:

  1. 首先对数组进行排序
  2. 第一个for循环固定遍历第一个元素
  3. 第二个循环移动两个指针的位置,移动逻辑为
if(nums[right]+nums[left]==target){
//找到的答案                  
}
//需要缩小和,只能移动right
if(nums[right]+nums[left]>target){
    right--;
}
//需要增大和,只能移动left
if(nums[right]+nums[left]<target){
    left++;
}
  1. 考虑去重复:最外层的一次遍历结束后,如果下一个要遍历的元素值相等,则跳过这一层遍历,因为上一次遍历已经把这个元素值为第一个值的所有情况全部遍历了;同理其他的也是
    整体代码如下
public List<List<Integer>> threeSum(int[] nums) {
        //一切的基础,双指针法要求必须排序
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<>();
        for(int i=0;i<nums.length-2;i++){

            //对于第一个元素的去重
            if(i>0&&nums[i-1]==nums[i]) continue;
            int target = -nums[i];
            
            int left = i+1;
            int right = nums.length-1;
            while(left<right){
                if(nums[right]+nums[left]==target){
                    result.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    left++;
                    right--;
                    //left去重
                    while(left<right&&nums[left]==nums[left-1]){
                        left++;
                    }
                    //right去重
                    while(left<right&&nums[right]==nums[right+1]){
                        right--;
                    }
                }
                //双指针移动的规则体现
                if(nums[right]+nums[left]>target){
                    right--;
                }
                if(nums[right]+nums[left]<target){
                    left++;
                }
            }
            
        }
        return result;
    }

LeetCode 18

题目链接:四数之和
升级版三数之和,还是使用双指针,循环再加一层

    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<>();
        for(int i=0; i<nums.length-3;i++){
            if(i>0&&nums[i]==nums[i-1])
                continue;
            for(int j=nums.length-1;j>=i+3;j--){
                if(j<nums.length-1&&nums[j]==nums[j+1])
                    continue;
                int left = i+1;
                int right = j-1;
                while(left<right){
                    if((long)nums[i]+nums[j]+nums[left]+nums[right]==target){
                        List<Integer> list = new ArrayList<>();
                        Collections.addAll(list, nums[i],nums[j],nums[left],nums[right]);
                        result.add(list);
                        left++;
                        right--;
                        while(nums[left]==nums[left-1]&&left<right){
                            left++;
                        }
                        while(nums[right]==nums[right+1]&&left<right){
                            right--;
                        }
                    }
                    if((long)nums[i]+nums[j]+nums[left]+nums[right]>target){
                        right--;
                    }
                    if((long)nums[i]+nums[j]+nums[left]+nums[right]<target){
                        left++;
                    }

                }
            }
        }
        return result;
    }
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容