算法 这个书,190页介绍了 这两个问题 。
2sum的意思是 在一组数中,找到 两个数的和为零。有多少个这样的组合。
3sum是 找 有多少三个数的组合 ,他们的和为零。
常规的for循环,问题的规模分别是 平方级别 和立方级别。
使用先排序,然后查找的方式,可以是 对数线性, 和 平方对数级别。
这样就快了一些。
这个转变是 ,把求一个组合的问题 ,转化成 已知一个数,找另一个数的问题。
查找就先排序(归并),再查找,二分。 这样就是 线性对数的速度。
排序的时间是 nlogn, 二分查找的时间是logn ,而者相加是
(n+1)logn 这比 n*n 要曲线更平缓一些。
TwoSumFast
public static int count(int []src){
//先排序
Arrays.sort(src);
int cnt = 0;
for(int i=0;i<src.length;i++){
//再用二分查找
if(BinarySearch.rank(- src[i]) > i)
cnt++;
}
return cnt;
}
ThreeSumFast
public static int count(int []src){
Arrays.sort(src);
int cnt = 0;
for(int i=0;i<src.length;i++){
for(int j=i+1;j<src.length;j++) {
if (BinarySearch.rank(-(src[i]+src[j])) > j)
cnt++;
}
}
return cnt;
}