数组、组合-[15] 三数之和

import java.util.Arrays;

/*
 * @lc app=leetcode.cn id=15 lang=java
 *
 * [15] 三数之和
 */

// @lc code=start
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        // 排序，主要用于处理存在相同项的问题，如果确定nums没有相同的项或返回结果中没有相同项，则可以排序
        Arrays.sort(nums);

        // 返回结果
        List<List<Integer>> ans = new ArrayList<>();

        // 因为每个结果有三个元素，所以只用循环到倒数第三个元素即可
        for(int i = 0; i < nums.length - 2; i++){
            // 跳过重复项，这里一定要跳过后面重复的项，否则结果中不会包括三个数中有两个或多个相同项的结果
            if(i > 0 && nums[i - 1] == nums[i])
                continue;
            
            // i 指向第一个确定的元素
            findAnotherNum(nums, i, ans);
        }

        return ans;
    }

    /***
     * 找到另外两个元素
     * @param nums 待筛选数组
     * @param ind 确定的一项
     * @param ans 保存结果的集合
     */
    private void findAnotherNum(int[] nums, int ind, List<List<Integer>> ans){
        // 根据题意这里另外两个数相加应该为 nums[ind]的相反数
        int twoNumSum = -nums[ind];
        
        // 左右指针开始遍历
        int le = ind + 1, ri = nums.length - 1;
        while(le < ri){
            // 当前左右指针对应元素所组成的和
            int sum = nums[le] + nums[ri];

            if(sum < twoNumSum){
                // 当前组成的和小于期望和则左指针向前移动
                le++;
            }else if(sum > twoNumSum){
                // 当前组成的和大于期望和则右指针向前移动
                ri--;
            }else{
                // 拼装符合元素
                List<Integer> sub = new ArrayList<>(3); 
                sub.add(nums[ind]);
                sub.add(nums[le]);
                sub.add(nums[ri]);
                
                // 加入返回结果
                ans.add(sub);

                // 直接跳入下一对
                le++;
                ri--;

                // 跳过重复的项
                while(le < ri && nums[le] == nums[le - 1])
                    le++;
                while(ri + 1 < nums.length && ri > le && nums[ri] == nums[ri + 1])
                    ri--;    
            }
        }
    }
}
// @lc code=end