三数之和
题目
给出一个有n个整数的数组S,在S中找到三个整数a, b, c,找到所有使得a + b + c = 0的三元组。
注意事项
在三元组(a, b, c),要求a <= b <= c。
结果不能包含重复的三元组。样例
如S = {-1 0 1 2 -1 -4}, 你需要返回的三元组集合的是:
(-1, 0, 1)
(-1, -1, 2)题解
先对数组排序,再进行遍历(注意跳过重复元素)。固定元素i,取双指针start,end分别指向i的后继区间[i+1,n-1]。定义sum = numbers[start] + numbers[i] + numbers[end],根据sum和0的大小关系移动指针的位置,关键还是要注意在移动指针时跳过重复元素。
public class Solution {
/**
* @param numbers : Give an array numbers of n integer
* @return : Find all unique triplets in the array which gives the sum of zero.
*/
public ArrayList<ArrayList<Integer>> threeSum(int[] numbers) {
Arrays.sort(numbers);
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
for (int i=0;i<numbers.length-2;i++)
{
if (i != 0 && numbers[i] == numbers[i-1])
{
continue;
}
int start = i+1;
int end = numbers.length-1;
while (start < end)
{
int sum = numbers[start] + numbers[i] + numbers[end];
if (sum == 0)
{
ArrayList<Integer> list = new ArrayList<Integer>();
list.add(numbers[i]);
list.add(numbers[start]);
list.add(numbers[end]);
result.add(list);
start++;
end--;
while (start < end && numbers[start] == numbers[start-1])
{
start++;
}
while (start < end && numbers[end] == numbers[end+1])
{
end--;
}
}
else if (sum > 0)
{
end--;
}
else
{
start++;
}
}
}
return result;
}
}
Last Update 2016.9.30