bzoj3444: 最後的晚餐(並查集+組合數學)

Hanks_o發表於2018-03-22

題目傳送門

解法:
並查集維護要挨在一起的人啊。
假設現在有tt段人要挨在一起。
然後這tt段人一共有t個人。
那麼其實只有n-t+tt個點在參與排序。
全排列就好。
然後每一段都可以倒過來。
所以最後要乘2^tt

判0其實蠻好判斷。
如果一個點的度>2的話那麼輸出0
或者當前這兩個點已經在同一個集裡面的話那肯定是衝突的啦(這是在我想了很多組資料猛然發現的)

程式碼實現:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
int fa[510000];int findfa(int x) {if(x!=fa[x])fa[x]=findfa(fa[x]);return fa[x];}const ll mod=989381;
bool v[510000];int sum[510000];
ll pow_mod(int b) {
    ll a=2;
    ll ans=1;while(b!=0) {if(b%2==1)ans=ans*a%mod;a=a*a%mod;b/=2;}
    return ans;
}
int s[510000];bool guding[510000];
int main() {
    memset(v,false,sizeof(v));int n,m;scanf("%d%d",&n,&m);int t=0;
    for(int i=1;i<=n;i++)fa[i]=i;memset(guding,false,sizeof(guding));
    for(int i=1;i<=m;i++) {
        int x,y;scanf("%d%d",&x,&y);
        int xx=findfa(x),yy=findfa(y);
        s[x]=y;if(s[y]==x)continue;
        if(xx==yy){printf("0\n");return 0;}
        if(xx!=yy)fa[xx]=yy;sum[x]++;sum[y]++;
        if(sum[x]>2||sum[y]>2) {printf("0\n");return 0;}
        if(v[x]==false) {v[x]=true;t++;}if(v[y]==false) {v[y]=true;t++;}
    }
    for(int i=1;i<=n;i++)fa[i]=findfa(i);
    memset(v,false,sizeof(v));int tt=0;
    for(int i=1;i<=n;i++) if(v[fa[i]]==false&&sum[i]!=0) {v[fa[i]]=true;tt++;}
    ll ans=1;for(ll i=2;i<=n-t+tt;i++) ans=ans*i%mod;
    printf("%lld\n",ans*pow_mod(tt)%mod);
    return 0;
}

相關文章