用中途相遇法的思想来解题。分别枚举两边,和直接暴力枚举四个数组比可以降低时间复杂度。
这里用到一个很实用的技巧:
求长度为n的有序数组a中的数k的个数num?
num=upper_bound(a,a+n,k)-lower_bound(a,a+n,k);
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> #include<map> #include<set> #include<vector> #include<algorithm> #include<stack> #include<queue> #include<cctype> #include<sstream> using namespace std; #define INF 1000000000 #define eps 1e-8 #define pii pair<int,int> #define LL long long int #define maxn 4005 int T,n,A[maxn],B[maxn],C[maxn],D[maxn],sum[maxn*maxn]; int main() { //freopen("in8.txt","r",stdin); //freopen("out.txt","w",stdout); scanf("%d",&T); while(T--) { scanf("%d",&n); for(int i=0;i<n;i++) { scanf("%d%d%d%d",&A[i],&B[i],&C[i],&D[i]); } int c=0; for(int i=0;i<n;i++) { for(int j=0;j<n;j++) { sum[c++]=A[i]+B[j]; } } sort(sum,sum+c); LL ans=0; for(int i=0;i<n;i++) { for(int j=0;j<n;j++) { ans+=upper_bound(sum,sum+c,-C[i]-D[j])-lower_bound(sum,sum+c,-C[i]-D[j]); //这一句是全篇的点睛之笔。 } } printf("%lld ",ans); if(T) printf(" "); } //fclose(stdin); //fclose(stdout); return 0; }