3-sum问题平方对数级别解法（algs4）

ThreeSum程序会统计一个文件中所有和为0的三整数元组的数量。ThreeSun用一个三重循环进行查找

for(int i=0;i<n;i++)
for(int j=i+1;j<n;j++)
for(int k=j+1;k<n;k++)
if(a[i]+a[j]+a[k])==0)
cnt++;
运行时间是关于数组大小的立方级别的

改进后的算法思路：

1.先将数组进行排序

2.将第三层循环判断改为二分查找

数量级由立方级别变为平方对数级别

                                                    


package _1_3linkedList;

import java.util.Arrays;
import java.util.Scanner;

import edu.princeton.cs.algs4.BinarySearch;

/*3-sum问题平方对数级别解法
*/
public class ThreeSumFast
{
public static int count(int [] a)
{/*计算和为0的三元组的数目*/
Arrays.sort(a);
int N=a.length;
int cnt=0;
for(int i=0;i<N;i++)
for(int j=i+1;j<N;j++)
if(BinarySearch.rank(-a[i]-a[j], a)>j) /*在有序数列里进行二分查找*/
cnt++; /*二分查找返回的数<j时不可重复计数*/
return cnt;
}
public static void main(String[] args)
{
Scanner sc=new Scanner(System.in);
int N=sc.nextInt();
int[] a=new int
;
for(int i=0;i<N;i++)
a[i]=sc.nextInt();
System.out.println(count(a));
}
}

7
1 2 3 4 -3 -2 -1
3