15. 三數之和

給定一個包含 n 個整數的數組 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。

注意:答案中不可以包含重複的三元組。

例如, 給定數組 nums = [-1, 0, 1, 2, -1, -4],

滿足要求的三元組集合爲:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

這道題首先需要先對數組排序,然後開始遍歷數組,對於每一個數都有一個目標值,即target = -nums[i],可以考慮從這個數後面的數left和最後一個數right 開始做判斷,這樣如果相加得到 target 則繼續判斷還有沒有另外兩個數滿足情況,直到left>=right,則這個數的所有情況判斷完畢,之後數組中和他相等的數則可以跳過。代碼如下:

public class Solution {
    IList<IList<int>> list = new List<IList<int>>();
    
    public IList<IList<int>> ThreeSum(int[] nums) {
        if(nums.Length < 3 || nums == null) return list;
        Array.Sort(nums);
        for(int i = 0; i < nums.Length - 2; i++){
            //因爲下面已經對重複數據做過處理,所以如果target值相等則跳過
            if(i > 0 && nums[i] == nums[i - 1]) continue;
            int left = i+1;     //左標誌位
            int right = nums.Length - 1;    //右標誌位
            int target = -nums[i];      //目標值
            while(left < right){
                if(nums[left] + nums[right] == target){
                    List<int> l = new List<int>();
                    l.Add(nums[i]);
                    l.Add(nums[left]);
                    l.Add(nums[right]);
                    list.Add(l);
                    //避免重複數據插入(已知兩個可以確定第三個)
                    while(left < right && nums[left] == nums[left + 1]){
                        left++;
                    }
                    while(left < right && nums[right] == nums[right - 1]){
                        right--;
                    }
                }
                //對於同一個target尋找下一組符合的值。
                if(nums[left] + nums[right] > target){
                    right--;  
                }else{
                    left++;
                }
            }
        }
        
        return list;
    }
}

代碼有點長,但很好理解。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章