LeetCode 15.三数之和

题目描述

链接：https://leetcode.cn/problems/3sum

---

解题思路

排序 + 双指针
注意点：去重

---

代码

public IList<IList<int>> ThreeSum(int[] nums)
{
    // 结果数组
    List<IList<int>> result = new List<IList<int>>();

    Array.Sort(nums);
    // 特殊情况处理
    if (nums.Length < 3) return result;
    for (int i = 0; i < nums.Length - 2; i++) 
    {
        if (nums[i] > 0) break; // 不会出现三数和为 0
        if (i > 0 && nums[i] == nums[i - 1]) continue;   // 去重
        // 双指针
        int head = i + 1;
        int trail = nums.Length - 1;
        
        while(trail > head)
        {
            // 三数之和
            int sum = nums[i] + nums[head] + nums[trail];
            // 对和进行判断
            if (sum == 0) 
            {
                result.Add(new List<int>(){nums[i], nums[head], nums[trail]});
                while (trail > head && nums[head] == nums[head + 1]) head++;    // 去重
                while (trail > head && nums[trail] == nums[trail - 1]) trail--; // 去重
                head++;
                trail--;
            }
            else if (sum > 0) trail--;
            else if (sum < 0) head++;
        }
    }
    return result;
}