uva1152 - 4 Values whose Sum is 0(hash或STL技巧ac)
阿新 • • 發佈:2018-07-29
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 namespacestd; 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)