1. 程式人生 > >[LeetCode] 15. 3Sum Java

[LeetCode] 15. 3Sum Java

pos arrays not AI 整體 第一次 排好序 == while

題目:Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

題意及分析:

求3Sum的思路如下:

  • 先整體排一次序
  • 然後遍歷數組,固定一個元素,對其後的元素求2Sum。

比如排好序以後的數字是 a b c d e f, 那麽第一次固定a, 在剩下的 b c d e f 中進行2sum, 完了以後第二次枚舉b, 只需要在 c d e f 中進行2sum,這樣就避免了重復。

public class Solution {
    public List<List<Integer>> threeSum(int[] nums) {   
        List<List<Integer>> res=new
ArrayList<>(); if (nums == null || nums.length < 3) return res; Arrays.sort(nums); int n=nums.length; for(int i=0;i<n-2;i++){ if (i > 0 && nums[i] == nums[i-1]) continue; int a=-nums[i]; int left=i+1;
int right=n-1; while(left<right){ if((nums[left]+nums[right])>a){ right--; }else if((nums[left]+nums[right])<a){ left++; }else{ List<Integer> tmp=new ArrayList<Integer>(); tmp.add(nums[i]); tmp.add(nums[left]); tmp.add(nums[right]); while (left < right && nums[left] == nums[left+1]) left++; while (left < right && nums[right] == nums[right-1]) right--; left++; right--; res.add(tmp); }; } } return res; } }

[LeetCode] 15. 3Sum Java