3Sum | Java最短代码实现

原题链接： 15. 3Sum
【思路】

蛮力算法，时间复杂度为O(n3)，无法ac。我们可以先排序O(nlogn)，我们知道2Sum是选定两指针从前后分别向中间扫，找到能找到就直接返回。与2Sum不同的是，3Sum要先选定一个数，再按照2Sum的思路找另两个数，即使找到了，也得继续找下去，不能直接返回。同时，要注意去重：

public List<List<Integer>> threeSum(int[] nums) {
ArrayList<List<Integer>> result = new ArrayList<List<Integer>>();
if (nums.length < 3) return result;
Arrays.sort(nums);
ArrayList<Integer> temp = null;
for (int i = 0; i < nums.length; i++) {
if (i > 0 && nums[i] == nums[i - 1]) continue;  //选定nums[i]为第一个数，并去重
int left = i + 1;
int right = nums.length - 1;
while (right > left) {
int sum = nums[i] + nums[left] + nums[right];
if (sum == 0) {
temp = new ArrayList<Integer>();
temp.add(nums[i]);
temp.add(nums[left]);
temp.add(nums[right]);
result.add(temp);
while (left < right && nums[left] == nums[left + 1]) left++;  //去重
while (left + 1 < right && nums[right] == nums[right - 1]) right--;
}
if (sum <= 0) left++;
else if (sum >= 0) right--;
}
}
return result;
}

311 / 311 test
cases passed. Runtime: 9
ms Your runtime beats 34.07% of javasubmissions.

欢迎优化！