L - Basketball Gym - 100741L---e二进制枚举

题意：

输入n,m。分别代表每个队人数，然后下面两行，第一行n个数，每个数字代表每个人能力值，第二行m个数，每个数字代表每个人能力值。求从n中选几个人，m中选几个人，能力值之和相等（即：两队势力平衡）的方法有多少种，注意：两队所选的人多数可以不同，只要能量值和相能就可以。

思路：

二进制枚举，把每种情况列举出来，然后看两个队和相等的情况有多少种，方法就有多少种。

第一遍的代码（超时）：  将两个组的所有自己子集情况分别存到两个数组里。然后比照和相等的情况，然而超时了。

#include <iostream>
#include <iomanip>
#include <cstring>
#include <cstdio>
#include <map>


using namespace std;
int ans; int k=0,l=0;
int N[18],M[18];
int aa[18],bb[18];


void subset(int n,int s,int chose,int *a)
{
    long long int tmp=0;
    for(int i=0;i<n;i++)
    {
        if(s&(1<<i))//如果两个位置上都是1则为真
       // printf("%d ",i);
          tmp+=a[i];
    }
    if(chose==1)
    {


       if(tmp)
      aa[k++]=tmp;
    }
    if(chose==0)
    {


       if(tmp)
      bb[l++] =tmp;


    }




}
void solve(int n,int m)
{
    for(int i=0;i<(1<<n);i++)
    {
        subset(n,i,1,N);
    }
    for(int i=0;i<(1<<m);i++)
    {
        subset(m,i,0,M);
    }


    for(int i=0;i<k;i++)
        for(int j=0;j<l;j++)
    {
        if(aa[i]==bb[j])
            ans++;


    }
    cout<<ans<<endl;


}
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;i++)
        cin>>N[i];
    for(int j=0;j<m;j++)
        cin>>M[j];
    solve(n,m);
}

AC 代码：（解决超时的问题）（应用map)

#include <iostream>
#include <iomanip>
#include <cstring>
#include <cstdio>
#include <map>


using namespace std;
 int k=0,l=0;
int N[18],M[18];
//int aa[18],bb[18];
map<long long,int> sa,sb;
map<long long,int> ::iterator it ;


void subset(int n,int s,int chose,int *a)// s 表示 2^n 种情况中的一种
{
    long long int tmp=0;
    for(int i=0;i<n;i++)
    {
        if(s&(1<<i))//如果两个位置上都是1则为真
       // printf("%d ",i);
          tmp+=a[i];
    }
    if(chose==1)
    {


       if(tmp)
       {
           if(sa.count(tmp))
            sa[tmp]++;
           else
            sa.insert(make_pair(tmp,1));
       }
    //  aa[k++]=tmp;
    }
    if(chose==0)
    {


       if(tmp)
       {
           if(sb.count(tmp))
            sb[tmp]++;
           else
            sb.insert(make_pair(tmp,1));
       }
    //  bb[l++] =tmp;


    }




}
void solve(int n,int m)
{
    long long int ans=0;
    for(int i=0;i<(1<<n);i++)
    {
        subset(n,i,1,N);
    }
    for(int i=0;i<(1<<m);i++)
    {
        subset(m,i,0,M);
    }


//    for(int i=0;i<k;i++)
//        for(int j=0;j<l;j++)
//    {
//        if(aa[i]==bb[j])
//            ans++;
//
//    }
  for ( it=sa.begin();it!=sa.end();it++)
  {
      if(sb.count(it->first))
      {
          ans+=((it->second) * sb[it->first]);
      }
  }
    printf("%lld\n",ans);


}
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;i++)
     scanf("%d",&N[i]);
    for(int j=0;j<m;j++)
         scanf("%d",&M[j]);
    solve(n,m);
}