牛客练习赛61 C.四个选项（并查集 DFS）

传送门

题意：

思路：

先把答案必须一样的用并查集放在一起
然后用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
*/