Problem:
Given an array of n integers nums and a target, find the number of index triplets i, j, k
with 0 <= i < j < k < n
that satisfy the condition nums[i] + nums[j] + nums[k] < target
.
For example, given nums = [-2, 0, 1, 3]
, and target = 2.
Return 2. Because there are two triplets which sums are less than 2:
[-2, 0, 1][-2, 0, 3]
Follow up:
Could you solve it in O(n2) runtime?General Analysis:
This problem is easy, but really tricky!It could force you to consider many detail you have not consider before.First, we need to understand it is difference with 3Sum problem.difference 1: the nums may still have duplicates, but we don't need to care about it, since the problme ask us to "find the number of index triplets i, j", which means as long as the elements are different(not value), we could record it.difference 2: it asks us to find all combinations smaller than "target", which is not the same as "equal", the way of couting answer should be different. I have made following mistakes to figure out the solution:
Wrong Solution 1:
public class Solution { public int threeSumSmaller(int[] nums, int target) { if (nums == null) throw new IllegalArgumentException("nums is null"); Listret = new ArrayList (); ret.add(0); Arrays.sort(nums); for (int i = 0; i < nums.length - 2; i++) { twoSumSmaller(nums, i + 1, target - nums[i], ret); } return ret.get(0); } private void twoSumSmaller(int[] nums, int i, int target, List ret) { int front = i; int end = nums.length - 1; while (front < end) { if (nums[front] + nums[end] < target) { ret.set(0, ret.get(0) + 1); end--; } else{ end--; } } }}
Mistakes Analysis:
Error cases:Input:[3,1,0,-2], 4Output:2Expected:3The above solution almost share the same code structure with 3Sum problem, excpet the condition to add "item" into "ret" list. And we don't need to care about duplicates things.However, the above logic of shirking the available range is wrong for this problem.Recall, for 3Sum Problem, we have,---------------------------------------------------if (nums[front] + nums[end] == target) { ... front++ (end--);}---------------------------------------------------We can move both front and end pointers. Since "nums[front] + nums[end] == target", whether we move front or end pointer, iff nums[front_new] != nums[front] (which is guaranteed), we still need to move end pointer. Thus, we actually need to move both pointers for possible available ranges, no possible answer would be skipped.But for this problem, it has great difference.Problemetic part:if (nums[front] + nums[end] < target) { ... end--;}For above code, the end may still in available range, but we discard it, thus we lose some answers."nums[front+1] + nums[end] < target" may be possible. Fix:if (nums[front] + nums[end] < target) { ... front++;}
Wrong Solution 2:
public class Solution { public int threeSumSmaller(int[] nums, int target) { if (nums == null) throw new IllegalArgumentException("nums is null"); Listret = new ArrayList (); ret.add(0); Arrays.sort(nums); for (int i = 0; i < nums.length - 2; i++) { twoSumSmaller(nums, i + 1, target - nums[i], ret); } return ret.get(0); } private void twoSumSmaller(int[] nums, int i, int target, List ret) { int front = i; int end = nums.length - 1; while (front < end) { if (nums[front] + nums[end] < target) { ret.set(0, ret.get(0) + 1); front++; } else{ end--; } } }}
Mistakes Analysis:
Error cases:Input:[3,1,0,-2], 4Output:2Expected:3Even though we fix the problem for moving pointer of this problem, we still face the other important problem.The problem with above solution is that: use the same counting method of 3Sum.Problemetic part:if (nums[front] + nums[end] < target) { ret.set(0, ret.get(0) + 1); front++;} If "nums[front] + nums[end] < target" is meet, all numbers from {nums[front] to nums[end-1]} must also meet the answer, with front fixed. thus we should have following counting method.ret.set(0, ret.get(0) + end - front);You may wander wheather it would incure duplicates in counting.Nope! since after we move front++, for ceratin "nums[i]", the same "nums[front]" would not appear again.
Analysis:
Ask yourself:This problem actually shares a lot of things with 3Sum. Like "binary search problem", this also represents a new genre of problem. Taking part of a sorting array. With one element was slected, for (int i = 0; i < nums.length - 2; i++) { twoSumSmaller(nums, i + 1, target - nums[i], ret);}How could you use the invariant "nums[front] + nums[end] meets certain condition" to control the searching in available ranges and couting right answer!!!It's an art!!! Right!
Solution:
public class Solution { public int threeSumSmaller(int[] nums, int target) { if (nums == null) throw new IllegalArgumentException("nums is null"); Listret = new ArrayList (); ret.add(0); Arrays.sort(nums); for (int i = 0; i < nums.length - 2; i++) { twoSumSmaller(nums, i + 1, target - nums[i], ret); } return ret.get(0); } private void twoSumSmaller(int[] nums, int i, int target, List ret) { int front = i; int end = nums.length - 1; while (front < end) { if (nums[front] + nums[end] < target) { ret.set(0, ret.get(0) + end - front); front++; } else{ end--; } } }}