【标签】 Array; Two Pointers
【个人分析】
老实交待,这个题卡半天,第一次做不会,抄别人的。过了很久,第二次做,还是不会……。好几次都是Time Limited Error。在看过正确答案之后,才知道是用的Two Pointers + sort 做的优化。
怎么优化? 简单说,就是通过 排序 + 跳过重复(利用Two Pointers) 来达到题目中避免 duplicates的要求。
核心思路: 我们给最终的由3个数组成的小数组叫 triplet。
1. 先锁定triplet中的第一个数字的index为 i (Line 12),
2. 然后我们希望从 [ nums[i + 1], nums[end]] 中找到两个数字,使得两个数字的和为 (0 - nums[i])
关键步骤:1. 在锁定第一个数字index的时候,跳过所有重复的数字 (Line 16处)。直到我们找到下一个不重复的数字作为triplet中的第一个数字
2. 在从[nums[i + 1], nums[end]]中找两个数字的时候,利用left pointer, right pointer。如果 left, right两个数的和比 twoSumTarget (Line 39)
小的话,只能通过将left pointer向右移动,来使得left + right的和更大一些,从而更接近twoSumTarget。
【一点心得】
在O(N^2)的算法中,考虑加入O(NlgN) 的排序操作。特别是遇到需要去除重复的时候,利用排序 + Two Pointers 来达到 HashSet的效果。
1 public classSolution {2 public List> threeSum(int[] nums) {3 List> result = new ArrayList>();4 int len =nums.length;5 if (len < 3) {6 returnresult;7 }8
9 Arrays.sort(nums);10
11 //for all number that can be the 1st number of triplet
12 for (int i = 0; i < len - 1; i++) {13 int firstNumber =nums[i];14
15 //skip all duplicated first number
16 if (i == 0 || firstNumber != nums[i - 1]) {17
18 int leftIndex = i + 1;19 int rightIndex = len - 1;20 int twoSumTarget = 0 -firstNumber;21
22 //try to find two numbers that sum up to twoSumTarget
23 while (leftIndex
27 result.add(Arrays.asList(firstNumber, nums[leftIndex], nums[rightIndex]));28 //skip duplicated nums[leftIndex]
29 while (leftIndex < rightIndex && nums[leftIndex] == nums[leftIndex + 1]) {30 leftIndex++;31 }32 //skip duplicated nums[rightIndex]
33 while (leftIndex < rightIndex && nums[rightIndex] == nums[rightIndex - 1]) {34 rightIndex--;35 }36 //move to next non-duplicates
37 leftIndex++;38 rightIndex--;39 } else if (twoSum
42 leftIndex++;43 } else{44 rightIndex--;45 }46 }47
48 }49 }50
51 returnresult;52 }53 }