15. 三数之和

image.png

排序后再遍历

执行用时:19 ms, 在所有 Java 提交中击败了99.26% 的用户 内存消耗:42.3 MB, 在所有 Java 提交中击败了63.22% 的用户

  1. class Solution {
  2. public List<List<Integer>> threeSum(int[] nums) {
  3. List<List<Integer>> resList = new ArrayList<>();
  4. if (nums == null || nums.length < 3) return resList;
  5. Arrays.sort(nums); // 排序
  6. for (int i = 0; i < nums.length - 2; i++) {
  7. if (nums[i] > 0) continue; // 如果第一个数大于0,因为后面的数也是大于0,三数之和不可能等于0,跳过
  8. if ((i > 0 && nums[i] == nums[i - 1])) continue; // 如果与上一个数相同,说明重复,跳过
  9. int left = i + 1, right = nums.length - 1;
  10. while (left < right) {
  11. if (nums[i] + nums[left] + nums[right] == 0) {
  12. resList.add(List.of(nums[i], nums[left], nums[right]));
  13. left++; right--;
  14. while (left < right && nums[left] == nums[left - 1]) left++; // 去除重复
  15. while (left < right && nums[right] == nums[right + 1]) right--; // 去除重复
  16. } else if (nums[i] + nums[left] + nums[right] < 0) {
  17. left++;
  18. } else {
  19. right--;
  20. }
  21. }
  22. }
  23. return resList;
  24. }
  25. }