1. 程式人生 > >uva1152 - 4 Values whose Sum is 0(hash或STL技巧ac)

uva1152 - 4 Values whose Sum is 0(hash或STL技巧ac)

namespace opera sort get ostream www 有序數組 size upper

題目大意:給定4個n(1 <= n <= 4000)元素集合A, B, C, D,要求分別從中選取一個元素a, b, c, d,使得a+b+c+d = 0,問有多少種選法。

method 1:

這裏用到一個很實用的技巧:

  求長度為n的有序數組a中的數k的個數num?

    num=upper_bound(a,a+n,k)-lower_bound(a,a+n,k);

#include <iostream>
#include <cstring>
#include <algorithm>
#include <map>

using namespace
std; const int maxn=4000+10; int T,n,A[maxn],B[maxn],C[maxn],D[maxn],sum[maxn*maxn];long long ans=0; int main() { ios::sync_with_stdio(false);cin.tie(0); cin>>T; while(T--){ ans=0; cin>>n; memset(sum,0,sizeof(sum)); for(int i=0;i<n;i++){ cin
>>A[i]>>B[i]>>C[i]>>D[i]; } int cnt=0; for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ sum[cnt++]=A[i]+B[j]; } } sort(sum,sum+cnt); for(int i=0;i<n;i++){ for(int j=0;j<n;j++){ ans
+=upper_bound(sum,sum+cnt,-C[i]-D[j])-lower_bound(sum,sum+cnt,-C[i]-D[j]); } } cout<<ans<<endl; if(T) cout<<endl; } return 0; }

method 2:運用hash的思想來做,類似於hdu1496 http://www.cnblogs.com/Fy1999/p/9355319.html

但需要加一個hash的結構體,否則會導致超時

#include <iostream>
#include <cstring>
#include <algorithm>
#include <map>

using namespace std;

const int maxn=4000+10;

int T,n,A[maxn],B[maxn],C[maxn],D[maxn];
long long ans=0;

struct HashMAP{
    static const int mask = 0x7fffff;
    int p[8388608],q[8388608];
    void Clear()
    {
        for(int i=0;i<=mask; ++ i) q[i]=0;
    }
    int& operator [] (int k){
        int i;
        for(i=k&mask; q[i]&&p[i]!=k;i=(i+1)&mask) ;
            p[i]=k;
        return q[i];
    }
}Hash;

int main()
{
    ios::sync_with_stdio(false);cin.tie(0);
    cin>>T;
    while(T--){
        Hash.Clear();//忘記清零導致TLE了。。。TLE不是WA有點坑啊。。。。。。
        ans=0;
        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++){
                Hash[A[i]+B[j]]++;
            }
        }

        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                ans+=Hash[-(C[i]+D[j])];
            }
        }
        cout<<ans<<endl;
        if(T) cout<<endl;
    }
    return 0;
}

uva1152 - 4 Values whose Sum is 0(hash或STL技巧ac)