题目链接:https://vjudge.net/problem/UVA-1152
题目大意:你输入四排数,每排代表一个集合,你可以从每个集合里边取出一个数,让其加和为0。
题目分析:这个题目最简单的做法就是n^4四个循环进行求解,但是这样就会爆时间,这样我们就可以采用中途相遇法,就是把a+b+c+d分解为a+b和c+d,然后再对a+b排序,再通过枚举c+d然后二分找到a+b里边符合题意的。(网上还有hash的帖子,这样可能速度会更快)
学到的东西:又熟悉了一遍二分的题目,了解了中间相遇法,多维问题可以把它分解为一个个低维的小问题。
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=16e6+10;
int m[4010][5];
int ab[maxn];
int cd[maxn];
int main()
{
int T;scanf("%d",&T);while(T--){
int n;scanf("%d",&n);
for(int i=0;i<n;i++)
for(int j=0;j<4;j++)
scanf("%d",&m[i][j]);
for(int i=0;i<n;i++)
for(int j=0;j<n;j++){//中间相遇法
ab[i*n+j]=m[i][0]+m[j][1];
cd[i*n+j]=m[i][2]+m[j][3];
}
int ans=0;
sort(ab,ab+n*n);
for(int i=0;i<n*n;i++){//枚举cd 找到对应的ab
int x=(-cd[i]);
int l=0,r=n*n-1;
while(l<=r){
int mid=(l+r)>>1;
if(ab[mid]>x) r=mid-1;
else if(ab[mid]<x) l=mid+1;
else{ //二分找到合适的
for(int k=mid;k>=0;k--){ //采用这种方法,要快于O n的做法
if(ab[k]==x) ans++;
else break;
}
for(int k=mid+1;k<=r;k++){
if(ab[k]==x) ans++;
else break;
}
break;
}
}
}
printf("%d\n",ans);
if(T) printf("\n");
}
}