题目
注意:三数之和属于双指针问题,二数之和属于哈希问题。
版本1代码
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
int n = nums.length;
if (n < 3) {
return res;
}
Arrays.sort(nums);
for (int i = 0; i < n - 2; ++i) {
if (nums[i] > 0) {
break;
}
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
int j = i + 1, k = n - 1;
while (j < k) {
int tmp = nums[i] + nums[j] + nums[k];
if (tmp == 0) {
res.add(new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[k])));
while (j + 1 < k && nums[j + 1] == nums[j]) {
++j;
}
while (k - 1 > j && nums[k - 1] == nums[k]) {
--k;
}
++j;
--k;
} else if (tmp < 0) {
++j;
} else {
--k;
}
}
}
return res;
}
}
版本2代码
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> res = new ArrayList<>();
if (nums.length < 3) {
return res;
}
Arrays.sort(nums);
int i = 0, j, k;
while (i < nums.length - 2) {
j = i + 1;
k = nums.length - 1;
while (j < k) {
if (nums[i] + nums[j] + nums[k] == 0) {
List<Integer> tmp = new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[k]));
res.add(tmp);
++j;
--k;
while (j < k && nums[j] == nums[j - 1]) {
++j;
}
while (j < k && nums[k] == nums[k + 1]) {
--k;
}
} else if (nums[i] + nums[j] + nums[k] > 0) {
--k;
} else {
++j;
}
}
++i;
while (i < nums.length && nums[i] == nums[i - 1]) {
++i;
}
}
return res;
}
}