題目大意：給定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; }

?

轉載于:https://www.cnblogs.com/Fy1999/p/9385926.html

總結

以上是生活随笔為你收集整理的uva1152 - 4 Values whose Sum is 0(hash或STL技巧ac)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。