8.5-237-uva1152-中途相遇法-二分法

合为0的四个值

给定4个n元素集合a,b,c,d，分别取一个元素出来使其和为0，有几种取法？

思路：

直接枚举需要四个循环 时间复杂度为O(n^4);

现在考虑中途相遇法

用两个循环遍历a+b 放入另一个数组sums1[]中 并排序 循环时间复杂度为O（n^2） 对sums[]中的n*n个元素排序为n^2*log(n^2)  总的复杂度为O（n^2*logn）

同理循环遍历c+d并用二分法查找-c-d 时间复杂度为n*n*log(n*2)

两个时间复杂度均为O（n^2*logn）

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=4000+5;
int a[maxn],b[maxn],c[maxn],d[maxn],sums[maxn*maxn];
int main(){
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int T;
	cin>>T;
	while(T--){
		int n;
		cin>>n;
		for(int i=0;i<n;i++)
		cin>>a[i]>>b[i]>>c[i]>>d[i];
		int k=0;
		for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
		sums[k++]=a[i]+b[j];
		sort(sums,sums+k); //排序 为二分查找作准备 
		int cnt=0;
		for(int i=0;i<n;i++)
		for(int j=0;j<n;j++){
			cnt+=upper_bound(sums,sums+k,-c[i]-d[j])-lower_bound(sums,sums+k,-c[i]-d[j]); //两函数之差返回查找值的个数 
		}
		cout<<cnt<<endl;
		if(T)  //最后一个情况 不需要换行 
		cout<<endl;
	}
	//system("pause");
	return 0;
}

下面是STL中的map函数方法 但在uva上超时

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<map> 
using namespace std;
const int maxn=4000+5;
int a[maxn],b[maxn],c[maxn],d[maxn],sums[maxn*maxn];
int main(){
	//freopen("in.txt","r",stdin);
	//freopen("out.txt","w",stdout);
	int T;
	cin>>T;
	while(T--){
		map<int,int> table;
		table.clear();
		int n;
		cin>>n;
		for(int i=0;i<n;i++)
		cin>>a[i]>>b[i]>>c[i]>>d[i];
		for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
		table[a[i]+b[j]]++;
		int cnt=0;
		for(int i=0;i<n;i++)
		for(int j=0;j<n;j++){
			cnt+=table[-c[i]-d[j]];  
		}
		cout<<cnt<<endl;
		if(T)  //最后一个情况 不需要换行 
		cout<<endl;
	}
	//system("pause");
	return 0;
}