1. 程式人生 > 實用技巧 >三個數之和

三個數之和


import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
*
* 給你一個包含 n 個整數的陣列nums,判斷nums中是否存在三個元素 a,b,c ,
* 使得a + b + c = 0 ?請你找出所有滿足條件且不重複的三元組。
*
* 注意:答案中不可以包含重複的三元組。
*
* 給定陣列 nums = [-1, 0, 1, 2, -1, -4],
* 滿足要求的三元組集合為:
* [
* [-1, 0, 1],
* [-1, -1, 2]
* ]
*
* 解題思路:
* 首先我們對陣列進行排序,然後定義兩個指標,
* 分別指向 當前遍歷索引的 下一個 和 陣列 最後一個元素的索引位置
* 在計算的過程當中,我們需要防止重複數字的重複計算。
*/
public class ThreeSum15 {

public List<List<Integer>> threeSum(int[] nums) {
//定義返回值
List<List<Integer>> res = new ArrayList<>();
int n = nums.length;
//如果傳入資料長度小於 3 直接 返回,題意要求三個數之和
if(n < 3) return res;
//先進行排序操作
Arrays.sort(nums);

//三個數的和
int sum = 0;
//遍歷整個陣列
for(int i = 0; i < n; i ++){
//如果當前這個數 num > 0 迴圈結束
if(nums[i] > 0) break;
//防止相同數字重複計算,造成結果相同
if(i > 0 && nums[i] == nums[i - 1]) continue;
//定義左右邊界
int l = i + 1;
int r = n - 1;
while(l < r){
//計算三個數的和
sum = nums[i] + nums[l] + nums[r];
if(sum == 0){
//滿足條件,裝入集合
res.add(Arrays.asList(nums[i],nums[l],nums[r]));
//防止相同數字重複計算,造成結果相同
while(l < r && nums[l] == nums[l + 1]){
l ++;
}
//防止相同數字重複計算,造成結果相同
while(l < r && nums[r] == nums[r - 1]){
r--;
}
l ++;
r --;
}else if(sum < 0){
l ++;
}else {
r --;
}
}
}
return res;
}

public static void main(String[] args) {
ThreeSum15 t = new ThreeSum15();
int[] nums = {-1, 0, 1, 2, -1, -4};
List<List<Integer>> lists = t.threeSum(nums);
System.out.println(lists);
}
}

題目來源:力扣官方