传送门
题意:
思路:
先把答案必须一样的用并查集放在一起
然后用cnt数组记录一下整合的每一堆的个数
然后dfs,判断所有情况即可(每次判断的是四个选项我可不可以选这个)
代码:
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <vector>
#include <math.h>
#include <map>
#include <queue>
#include <set>
#include <stack>
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define fi first
#define se second
#define all(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define debug(x) cout<<x<<endl
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
typedef long long ll;
using namespace std;
const int MAXN=1e5+50;
const int inf=0x3f3f3f3f;
const int mod=1e9+7;
//::iterator it;
struct sa{
int x;
int y;
}p[1050];
int cnt[15];
int kkk[15];
int a[10];
ll ans=0;
void dfs(int gg){
if(gg==0){
ans++;
return ;
}
int f=0;
for(int i=1;i<=4;i++){
if(a[i]>=cnt[gg]){
f=1;
a[i]-=cnt[gg];
dfs(gg-1);
a[i]+=cnt[gg];
}
}
if(f==0)return;
}
int k[20];
int find(int x)
{
if(k[x]!=x)
k[x]=find(k[x]);
return k[x];
}
void join(int x,int y)
{
int x1=find(x),y1=find(y);
k[y1]=x1;
}
int main()
{
std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
int na,nb,nc,nd,m;
for(int i=1;i<=4;i++)cin>>a[i];
cin>>m;
for(int i=1;i<=m;i++)cin>>p[i].x>>p[i].y;
for(int i=1;i<=12;i++)k[i]=i;
for(int i=1;i<=m;i++){
int ev,ew;
ev=find(p[i].x);
ew=find(p[i].y);
if(ev!=ew){
join(ev,ew);
}
}
for(int i=1;i<=12;i++)k[i]=find(i);
int sum=0;
for(int i=1;i<=12;i++)kkk[k[i]]++;
int gg=0;
for(int i=1;i<=15;i++)if(kkk[i]!=0){
cnt[++gg]=kkk[i];
}
sort(cnt+1,cnt+gg+1);
//for(int i=1;i<=gg;i++)cout<<cnt[i]<<endl;
sort(a+1,a+5);
dfs(gg);//一共gg堆
cout<<ans<<endl;
return 0;
}
/*
3 3 3 3 2
1 2
2 3
*/