數組中相加和為0的三元組


給出一個有n個元素的數組S,S中是否有元素a,b,c滿足a+b+c=0?找出數組S中所有滿足條件的三元組。
注意:
三元組(a、b、c)中的元素必須按非降序排列。(即a≤b≤c)
解集中不能包含重復的三元組。
例如,給定的數組 S = {-10 0 10 20 -10 -40},解集為(-10, -10, 20),(-10, 0, 10)
0 <= S.length <= 1000
示例1
輸入:[0]
返回值:[]

示例2
輸入:[-2,0,1,1,2]
返回值:[[-2,0,2],[-2,1,1]]

示例3
輸入:[-10,0,10,20,-10,-40]
返回值:[[-10,-10,20],[-10,0,10]]

public ArrayList<ArrayList<Integer>> threeSum(int[] num) {

        ArrayList<ArrayList<Integer>> result = new ArrayList<>();

        if (num == null) {
            return result;
        }
        //排序
        Arrays.sort(num);
        int sum, left, right;

        for (int i = 0; i < num.length - 2; i++) {
            //避免重復計算
            if (i != 0 && num[i] == num[i - 1]) {
                continue;
            }
            left = i + 1;
            right = num.length - 1;
            /**
             * 固定一個數,從后面的數中選出兩個數,因為數組是有序的,所以可以
             * 用兩個數組下標left和right,left指向當前元素的后一個位置,right指向數組的最后一個位置
             * 三個數相加的和等於0時,加入解集;
             * 小於0時,把left往右邊移動;
             * 大於0時,把right往左邊移動;
             */
            while (left < right) {
                sum = num[left] + num[right];
                if (sum + num[i] == 0) {
                    ArrayList<Integer> solution = new ArrayList<>();
                    solution.add(num[i]);
                    solution.add(num[left]);
                    solution.add(num[right]);
                    result.add(solution);
                    left++;
                    right--;
                    //跳過重復元素計算
                    while (left < right && num[left] == num[left - 1]) {
                        left++;
                    }
                    //跳過重復元素計算
                    while (left < right && num[right] == num[right + 1]) {
                        right--;
                    }
                } else if (sum + num[i] < 0) {
                    left++;
                } else {
                    right--;
                }
            }
        }
        return result;
    }


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM