1152 - 4 Values whose Sum is 0

紫书上叫中途相遇法,还有一个名字感觉更加妥帖一点,叫:折半枚举。          有时候,当问题的规模较大时,无法枚举所有元素的组合,但能够枚举一半的元素组合,此时,将问题拆成两半后分别枚举,再合并他们的结果这一方法往往非常有效。

两重循环加二分,总复杂度为n^2logn 

这里值得一提的是对集合CD的存储方式,我是用了一个有序数组,也可以用其他方式。

#include<bits/stdc++.h>
using namespace std;
const int max_n =4005;
int n,T;
long long A[max_n],B[max_n],C[max_n],D[max_n],CD[max_n*max_n];
void solve(){
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        CD[i*n+j] = C[i] + D[j];
    sort(CD,CD+n*n);
    long long res = 0;
    for(int i=0;i<n;i++)
    for(int j=0;j<n;j++){
        int cd = -(A[i]+B[j]);
        res+=upper_bound(CD,CD+n*n,cd) - lower_bound(CD,CD+n*n,cd);
    }
    printf("%lld\n",res);
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        for(int i=0;i<n;i++){
            scanf("%lld",&A[i]);
            scanf("%lld",&B[i]);
            scanf("%lld",&C[i]);
            scanf("%lld",&D[i]);
        }
        solve();
        if(T) printf("\n");
    }
    return 0;
}



你可能感兴趣的:(ACM,uva)