leetcode刷题日记1：两数之和-三数之和-四数之和

抓狂中

---

前言

两数之和-三数之和-四数之和三个题算是一系列的，主要掌握双指针法，配合一些剪枝操作，解题思路就有了，以后n个数之和也能轻松解决；

一、两数之和

题目

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

思路

循环嵌套遍历也可，但是时间复杂度是O(n²)，采用HashMap，遍历一遍时把元素加进map里，然后找targer-当前元素值是否在map里即可；
注：由于可能存在相同值，所以要先检查所要找的值是否在map里，在把当前值加进map，如nums=[…3,3,…],target=6,如果在把第一个3先加进map，则判断map里有无3会返回两个相同的位置值。

代码

class Solution {
    public int[] twoSum(int[] nums, int target) {
        Map<Integer,Integer> m = new HashMap<Integer,Integer>();//map的键是元素，值是位置

        int len=nums.length;
        for(int i=0;i<len;i++)
        {
            if(m.containsKey(target-nums[i]))
            return new int[]{m.get(target-nums[i]),i};

            //如果有相同元素，且target的值等于这两元素之和，则必须在判断完存在之后才能将当前遍历的元素加入map中
             m.put(nums[i],i);//遍历过程中把每个元素都添加进map中
        }
        return new int[]{};
    }
}

二、三数之和

题目

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

思路

双指针法

代码

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> l = new ArrayList<>();
        
        for(int begin=0;begin<nums.length-2;begin++ )
        {
            if(nums[begin]>0)
                break;
            if(begin>0&&nums[begin]==nums[begin-1])
                continue;//如果当前项和前一项相等，说明重复了，跳过
            int left=begin+1,right=nums.length-1,sum=0;
            while(left<right)
            {
                sum=nums[left]+nums[right]+nums[begin];
                if(sum==0)
                    {
                        l.add(new ArrayList<Integer>(Arrays.asList(nums[begin],nums[left],nums[right])));
                        while(left<right&&nums[left]==nums[++left]);
                        while(left<right&&nums[right]==nums[--right]);
                    }
                else if(sum<0)
                    while(left<right&&nums[left]==nums[++left]);
                else
                    while(left<right&&nums[right]==nums[--right]);
            }
        }
        return l;
    }
}

三、四数之和

题目

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target
你可以按 任意顺序 返回答案 。

思路

同样是双指针法，不过多了很多剪枝

代码

class Solution {
    //双指针法
    public List<List<Integer>> fourSum(int[] nums, int target){
        List<List<Integer>> res = new ArrayList<List<Integer>>();//返回的答案
        if(nums==null||nums.length<4)//如果数组为空或者长度小于4，返回空数组
            return res;
        int len=nums.length;
        Arrays.sort(nums);
        for(int i=0;i<len-3;i++)
        {
            //剪枝
            if(i>0&&nums[i]==nums[i-1])//如果和上一个元素的值相等，表示该值已经判断过了
                continue;
            if(nums[i]+nums[i+1]>target-(nums[i+2]+nums[i+3]))//四个元素和如果超出了int的范围，改成减法活用long强转，下同
                break;//如果前四个元素相加已经比target大了，则后面肯定没有小于等于target的数了，结束循环
            if(nums[i]+nums[len-1]<target-(nums[len-2]+nums[len-3]))
                continue;//如果第一个和最后三个数相加比target小，则第一个数肯定不满足，直接从下一个数开始
            for(int j=i+1;j<len-2;j++)
            {
                //剪枝方法同上
                if(j>i+1&&nums[j]==nums[j-1])
                    continue;
                if(nums[i]+nums[j]>target-(nums[j+1]+nums[j+2]))
                    break;
                if(nums[i]+nums[j]<target-(nums[len-2]+nums[len-1]))
                    continue;
                int left=j+1,right=len-1;
                //双指针法
                while(left<right)
                {
                    int sum1 = nums[i]+nums[j];
                    int sum2 = target-(nums[left]+nums[right]);
                    if(sum1<sum2)
                        while(left<right&&nums[left]==nums[++left]);//跳过重复元素
                    else if(sum1>sum2)
                        while(left<right&&nums[right]==nums[--right]);//跳过重复元素
                    else
                    {
                        res.add(Arrays.asList(nums[i],nums[j],nums[left],nums[right]));
                        while(left<right&&nums[left]==nums[++left]);
                            //另一种麻烦的写法
                            // while(left<right&&nums[left+1]==nums[left])
                            //    left++;
                            //left++;//不加这一步跳不出循环，永远卡在最后一个相等的left
                            // 通过在nums[++left]做自加，包正最后能指项下一个不同的元素
                        while(left<right&&nums[right]==nums[--right]);
                            //另一种麻烦的写法
                            // while(left<right&&nums[right-1]==nums[right])
                            //     right--;
                            //right--;//不加这一步跳不出循环，永远卡在最后一个相等的right
                            //改进方法同上
                    }
                }
            }
        }

        return res;
    }
}