LintCode 四数之和题目分析代码

题目

给一个包含n个数的整数数组S,在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。

注意事项

四元组(a, b, c, d)中,需要满足a <= b <= c <= d

答案中不可以包含重复的四元组。

样例 例如,对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为:

(-1, 0, 0, 1)

(-2, -1, 1, 2)

(-2, 0, 0, 2)

分析

跟三数之和思路一样,多一层循环而已

代码

public class Solution {
    /**
     * @param numbers : Give an array numbersbers of n integer
     * @param target : you need to find four elements that's sum of target
     * @return : Find all unique quadruplets in the array which gives the sum of
     *           zero.
     */
    public ArrayList<ArrayList<Integer>> fourSum(int[] num, int target) {
        if(num == null || num.length <4)
            return new ArrayList<>();
            
            
        ArrayList<ArrayList<Integer>> rst = new ArrayList<ArrayList<Integer>>();
        Arrays.sort(num);
        
        for(int i=0;i<num.length-3;i++) {
            if(i!=0 && num[i] ==num[i-1])
                continue;
            
            for(int j=i+1;j<num.length-2;j++) {
                if(j!=i+1 && num[j] ==num[j-1])
                    continue;
                
                int left = j+1;
                int right = num.length-1;
                
                while(left < right) {
                    int sum = num[i] + num[j] + num[left] + num[right];
                    if(sum == target) {
                        ArrayList<Integer> temp = new ArrayList<>();
                        temp.add(num[i]);
                        temp.add(num[j]);
                        temp.add(num[left]);
                        temp.add(num[right]);
                        rst.add(temp);
                        left++;
                        right--;
                        while(left<right && num[left] == num[left-1])
                            left++;
                        while(left<right && num[right] == num[right+1])
                            right--;
                    } else if(sum < target)
                        left++;
                    else
                        right--;
                    
                }
            }
        }
        return rst;
                
    }
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券