合为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;
}