POJ2785-4 Values whose Sum is 0(二分基础）

题目链接

题意：
给定四个长度均为n的数字序列，从这四个序列中各任选一个数字，设分别为a,b,c,d，求a+b+c+d=0有多少种可能的方案。（各个序列间数字的选择是相互独立的）

题目条件：
n<=4000,|Xi|<=2^28。（Xi表示选择的数字）

分析：
因为之前在挑战程序设计竞赛书上看过原题，但是解答当时并未全部理解，这次碰到了就好好分析一下。首先看完题目，想到暴力遍历四层的for循环，O(n^4)的复杂度是绝对接受不了的，那么就要考虑下，如何降低时间复杂度。由于我们选择的四个不同序列的数之间是相互独立的，那么可以先用两层for循环把a+b的可能性全部枚举出来存到n * n的数组里面，然后我一开始解题的以为接下来的遍历层序可以任意改变，结果发现自己搞了个假算法，我是先外层遍历一遍 n * n数组，然后内层遍历第三个数组，再最内层二分查找排好序的第四个数组的d值，因为a+b+c已经枚举出，d值假定存在那么就是-(a+b+c)，如果能二分找到就说明是一种选择方案，否则枚举下一个。而实际我的复杂度并没有降很多，由于我的天真，外层循环看似一层实则还是n ^2的复杂度，所以合起来就是O(n ^3logn)的复杂度，交了一发，果断超时，最后发现，我应该把遍历n * n的数组操作放到循环内部，先两层循环遍历第三，四个数组，再内部二分查找n * n的数组，这样就真正达到了大大降低复杂度的目的，复杂度就是O(n ^2logn ^2)=O(2*n ^2logn)–>O(n ^2logn)，还有就是利用upper_bound-lower_bound等于查询数在数组中个数的特性，得到能有的多组方案数，其他要注意就是需要scanf读入。

主要代码

int n;
void solve(){
	vector<vector<int>> s(4,VI(n));
	for(int i=0;i<n;i++){
		for(int j=0;j<4;j++)
		scanf("%d",&s[j][i]); 
	}
	VI b;
	for(int i=0;i<s[0].size();i++){
		for(j=0;j<s[1].size();j++){
			b.push_back(s[0][i]+s[1][j]);
		}
	}
	sort(ALL(b));
	ll ans=0;
	for(int i=0;i<s[3].size();i++){
		for(int j=0;j<s[2].size();j++){
			ans+=upper_bound(ALL(b),-(s[3][i]+s[2][j]))-lower_bound(ALL(b),-(s[3][i]+s[2][j]));
		}
	}
	printf("%lld\n",ans);
}
int main(){
//	ios::sync_with_stdio(false);
//	cin.tie(0);
	while(cin>>n)
	solve();
	return 0;
}

等刷的多了，弄一个合集。。。