算法第四版第一章练习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];
		//全部加上最小值的绝对值,保证最小值为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);
	}


猜你喜欢

转载自blog.csdn.net/KingYMxe/article/details/77882808