给定一个数组,有正有负,列出其中三个数和为零的组合
public class FindThree {
public static void main(String[] args) {
int[] arr = { 2, 1, -1, -9, 10, 8, 7 };
System.out.println(process(arr));
}
private static String process(int[] keys) {
Arrays.sort(keys);
print(keys);
HashSet<Vector<Integer>> set = new HashSet<>();
for (int i = 0; i < keys.length; i++) {
int a = keys[i];
int j = i + 1;
int k = keys.length - 1;
while (j < k) {
int sum = a + keys[j] + keys[k];
if (sum < 0) {
j++;
} else if (sum > 0) {
k--;
} else {
Vector<Integer> vector = new Vector<>(3);
vector.add(a);
vector.add(keys[j]);
vector.add(keys[k]);
set.add(vector);
j++;
k--;
}
}
}
return set.toString();
}
private static void print(int[] keys) {
for (int key : keys) {
System.out.print(key + " ");
}
System.out.println();
}
}
输出结果:
-9 -1 1 2 7 8 10
[[-9, -1, 10], [-9, 2, 7], [-9, 1, 8]]
查找了一下网上的答案,通过java进行了实现。如上:
1、对数组进行排序
2、进行遍历:寻找存在a的组合。
如果sum小于0,则将第二个数后移一位,使sum增大;
如果sum大于0,则将第三个数前移一位,使sum减小;
当sum等于0时,则得到对应组合,此时j++,k--,寻找 当前a 存在的下一个组合
时间复杂度 O(n^2)