Round 3 D - Balance POJ - 1837 -树形DP

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
#define D(v) cout<<#v<<" "<<v<<endl
#define mem(s,t) memset(s,t,sizeof(s))
const int MAXN =25;
int C,G;
int c[MAXN],g[MAXN];
set<int> s,S;
map<int,int> m,M;
typedef set<int>::iterator It;
int main(){
cin>>C>>G;
for(int i=0;i<C;i++){
cin>>c[i];
}
for(int j=0;j<G;j++){
cin>>g[j];
}
M[0]=1;
for(int i=0;i<G-1;i++){
m=M;
s=S;
M.clear();
S.clear();
s.insert(0);
for(int j=0;j<C;j++){
int val =g[i]*c[j];
for(It it=s.begin();it!=s.end();it++){
//D(*it);
S.insert(*it+val);
M[*it+val] += m[*it];
}
}
}
int cnt=0;
int Ans=0;
for(int i=0;i<C;i++){
Ans=c[i]*g[G-1];
cnt+=M[-Ans];
}
cout<<cnt<<endl;
return 0;
}