給定一個包含 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;
}
}
代碼有點長,但很好理解。