zoukankan      html  css  js  c++  java
  • uva1152

    题目大意:给定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;
    }
  • 相关阅读:
    JAVA学习日报 12/15
    JAVA学习日报 12/14
    JAVA学习日报 12/13
    如何在idea上配置meaven和tomcat!(idea2020)
    关于win10共存多个版本jdk,并如何进行最简单的切换
    Java面试知识点
    一些基本html标签的使用案例
    线性代数复习笔记
    考前加分项
    Javaweb学习12.18
  • 原文地址:https://www.cnblogs.com/Fy1999/p/9385926.html
Copyright © 2011-2022 走看看