题目大意:给定一个数组,求取其中和为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];
//全部加上最小值的绝对值,保证最小值为0
for(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);
}