算法刷题:四数之和

.

题目链接

四数之和

题目详情

算法原理

题目要求的是a+b+c=target
这道题我们可以参考一下三数之和
三数之和中,我们固定的是其中一个数,四数之和中,我们则需要固定两个数
即
a+b+c=target的时候固定d,找到a+b+c=target-d
在求a+b+c=target-d的时候,固定c,找到a+b=target-d-c;
然后利用两数之和求解
注意:
当数据进行加减操作的时候,可能会发生溢出,因此,我们需要在计算a+b=target-d-c;的时候,将数据转换为long

我的答案

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>>list = new ArrayList<>();
        Arrays.sort(nums);
        int n = nums.length;
        for(int a = 0;a<n;){
            for(int b = a+1;b<n;){
               long tar = (long)target-nums[a]-nums[b];
                int left = b+1;
                int right = n-1;
                while(left<right){
                    long sum = nums[left]+nums[right];
                    if(sum<tar) left++;
                    else if(sum>tar) right--;
                    else{
                        list.add(new ArrayList<Integer>(Arrays.asList(nums[a],nums[b],nums[left],nums[right])));
                        left++;
                        right--;
                        while(left<right&&nums[left]==nums[left-1]) left++;
                        while(left<right&&nums[right]==nums[right+1]) right--;
                    }
                }
                b++;
                while(b<n&&nums[b]==nums[b-1]) b++;
            }
            a++;
            while(a<n&&nums[a]==nums[a-1]) a++;
        }
        return list;
    }
}