UOJ 【UR #12】密码锁

密码锁

Description

http://uoj.ac/problem/181

Data Constraint

$1$≤$n$≤$38$, $0$≤$m$≤$19$,$0$≤$w_i$≤$10^4$

Solution

由于定向后原图是一个竞赛图，进行强连通分量缩点后，一定是一个拓扑序唯一的$DAG$，此$DAG$的点数的期望值就是答案。

不难发现，此$DAG$可以分成两个集合$S$,$T$使得$S$和$T$之间的所有连边都是$S$连向$T$的，可以发现，此时的答案就是合法的划分方案数+$1$。

于是我们可以将答案写成$\sum_{S,T}\Pi_{x\in S\ y\in T}P_{x,y}$其中$P_{x,y}$是$x$指向$y$的概率。

当$m$=$0$时，我们可以得出$Ans=1+\sum_{i=1}^{n-1}C_{n}^i{1\over 2}^{i(n-i)}$。

那当$m$不等于$0$时，我们可以对特殊边组成的每个弱连通分量分开考虑。
对于一个弱连通分量$K$，我们用$2^{|K|}$来枚举每个点是属于$S$集还是属于$T$，若一条特殊边连接的两个端点属于不同的集合，则需要乘上这条边定向的概率。
对于一个弱连通分量我们可以求出一个$f$，$f_i$表示此弱连通分量内有$i$个点属于$S$集的总概率之和（不算上非特殊边）。
再设一个$ans_i$表示做到当前弱连通分量时总共有$i$个点属于$S$集时的概率。
我们每做一个弱连通分量就将得到的$f$和$ans$卷积以更新$ans$。

最后答案就为$Ans=1+\sum_{i=1}^{n-1}ans_i{1\over 2}^{i(n-i)}$
可以证明时间复杂度的上限为$O(2^{m+1}n+n^2)$

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

#define fo(i,j,l) for(int i=j;i<=l;++i)
#define fd(i,j,l) for(int i=j;i>=l;--i)

using namespace std;
typedef long long ll;
const ll N=49,M=2e6,mo=998244353;

ll f[N],g[N];
int x[N],y[N],c[N],d[N],m2[N];
int sta[N],fa[N];
int n,m,i,o,p;

inline ll ksm(ll o,ll t)
{
    ll y=1;
    for(;t;t>>=1,o=o*o%mo)
    if(t&1)y=y*o%mo;
    return y;
}

int gf(int a)
{return fa[a]==a?a:fa[a]=gf(fa[a]);}

inline int bits(int o)
{return (!o)?0:(o&1)+bits(o>>1);}

int main()
{
    freopen("random.in","r",stdin);
    freopen("random.out","w",stdout);
    cin>>n>>m;
    fo(i,1,n)fa[i]=i;
    fo(i,1,m){
        scanf("%d%d%d",&x[i],&y[i],&c[i]);
        if(gf(x[i])!=gf(y[i])){
            fa[fa[x[i]]]=fa[y[i]];
        }
    }
    m2[1]=1;
    fo(i,2,n+1)m2[i]=m2[i-1]<<1;
    f[0]=1; ll ny=ksm(10000,mo-2);
    fo(i,1,n)if(fa[i]){
        int v=0;
        fo(l,0,n)g[l]=0;
        fo(l,1,n)if(gf(i)==gf(l))d[++v]=l;
        fo(l,1,v)fa[d[l]]=-1;
        fo(l,0,m2[v+1]-1){
            ll tot=1; int tj=0;
            fo(j,1,v)sta[d[j]]=(l&m2[j])>0;
            fo(j,1,m)if(fa[x[j]]+fa[y[j]]==-2){
            if(sta[x[j]]==1&&sta[y[j]]==0)tot=tot*2*c[j]%mo*ny%mo;
            else if(sta[x[j]]==0&&sta[y[j]]==1)tot=tot*2*(10000-c[j])%mo*ny%mo;
            }
            int gs=bits(l);
            g[gs]=(g[gs]+tot)%mo;
        }
        fd(l,n,0){
            ll op=0;
            fo(j,0,l)op=(op+f[j]*g[l-j])%mo;
            f[l]=op;
        }
        fo(l,1,v)fa[d[l]]=0;
    }
    ll ans=1;
    fo(i,1,n-1)ans=(ans+f[i]*ksm((mo+1)>>1,i*(n-i)))%mo;
    ans=ans*ksm(10000,n*(n-1))%mo;
    cout<<ans; 
}