算法第四版第一章练习1.4.15 3_sum的O(n²)解法

题目大意：给定一个数组，求取其中和为0的3个数的组数。时间复杂度要求O(n²)

我这个应该可以吧。网上也找不着答案。。

重点就是用一个count辅助数组把数组中的值给存储起来，方便索引，注意存储时数组会越界，这里我选择把所有值都给加上数组的最小值的绝对值。

public static void main(String []args){int []a= {-2,-1,2,3};int min=9999;//取出其中的最小值，防止接下来存储进count数组时数组越界for(int i=0;i<a.length;i++)if(a[i]<min)min=a[i];//全部加上最小值的绝对值，保证最小值为0for(int i=0;i<a.length;i++)a[i]+=Math.abs(min);int []count=new int[5000];//把count数组的下标当做数组中每个数字的值，能够迅速索引出是否有这个值和其下标for(int i=0;i<a.length;i++) {count[a[i]]=i;}int ans=0;for(int i=0;i<a.length;i++) {for(int j=i+1;j<a.length;j++) {//因为每个值都加上了最小值，所以三个数的和为0在后面就相变成了三个数的和为最小值的绝对值的3倍if(3*Math.abs(min)-a[i]-a[j]>=0&&count[3*Math.abs(min)-a[i]-a[j]]>j)ans++;}}System.out.println(ans);}