【LeetCode】#15三數之和(3Sum)
阿新 • • 發佈:2018-12-03
【LeetCode】#15三數之和(3Sum)
題目描述
給定一個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。
注意:答案中不可以包含重複的三元組。
示例
例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4],
滿足要求的三元組集合為:
[
[-1, 0, 1],
[-1, -1, 2]
]
Description
Given an array nums of n integers, are there elements a, b, c in nums 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.
Example
Given array nums = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]
解法
class Solution{ public List<List<Integer>> threeSum(int[] nums){ Arrays.sort(nums); int iNum = Integer.MAX_VALUE; int jNum; int kNum; List<List<Integer>> res = new ArrayList<>(); for(int i=0; i<nums.length && nums[i]<0; i++){ if(iNum==nums[i]){ continue; } iNum = nums[i]; int j = i + 1; int k = nums.length-1; while(j<k){ int sum = nums[i] + nums[j] + nums[k]; if(sum>0){ k--; }else if(sum<0){ j++; }else{ List<Integer> list = new ArrayList<>(); list.add(nums[i]); list.add(nums[j]); list.add(nums[k]); res.add(list); jNum = nums[j]; do{ j++; if(j>=k){ break; } }while(jNum==nums[j]); kNum = nums[k]; do{ k--; if(j>=k){ break; } }while(kNum==nums[k]); } } } return res; } }