三数之和问题可以通过一定的方法转化成两数之和问题来求解;
解题思路:
固定一个数字nums[i],使其他两个数的相加等于-nums[i],即nums[j] + nums[k] = -nums[i]。
但是这里要考虑到结果不能重复,处理的办法是将数组nums先进行从小到大的排序。
因为对数组nums进行了排序,所以有一种改进方法,固定了nums[i],让[ j ]从[ i+1 ]开始而让[ k ]从末尾开始,如果nums[j] + nums[k] > -nums[i],说明左边数值太大,k应该减小来减小数值,而如果nums[j] + nums[k] < -nums[i]则表明左边的数值太大,j应该增大来增大数值。
代码如下:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> result = new LinkedList<>();
if(nums == null || nums.length < 3)
return result;
Arrays.sort(nums);
for(int i = 0; i < nums.length-2; i++){
int j = i + 1;
int k = nums.length - 1;
while(j < k){
if(nums[j] + nums[k] == -nums[i]){
List<Integer> list = new ArrayList<>(3);
list.add(nums[i]);
list.add(nums[j]);
list.add(nums[k]);
result.add(list);
j++;
k--;
while(j < k && nums[j] == nums[j-1])
j++;
while(j < k && nums[k] == nums[k+1])
k--;
}
else if(nums[j] + nums[k] > -nums[i]){
k--;
while(j < k && nums[k] == nums[k+1])
k--;
}
else{
j++;
while(j < k && nums[j] == nums[j-1])
j++;
}
}
while(i < nums.length - 2 && nums[i] == nums[i+1])
i++;
}
return result;
}
}
当然,代码执行效率不是特别高,等刷完了一遍LeetCode再说吧,思路是这个样子。