HDU 6064 RXD and numbers(生成树计数+行列式)

来源:互联网 发布:公司记账软件 知乎 编辑:程序博客网 时间:2024/05/21 09:39

Description
一个长度为n的序列A满足
1.1<=A[i]<=m
2.A[1]=A[n]=1
3.对任意1<=x<=m,至少存在一个1<=i<=n使得A[i]=x
4.对任意1<=x,y<=m,满足A[i]=x,A[i+1]=y的i的数量是D[x][y]
给出D,求满足条件的A序列数量
Input
多组用例,每组用例首先输入一整数m,之后输入一m*m矩阵D,以文件尾结束输入(0<=D[i][j] < 500,1<=m<=400,2<=n=sum{D[i][j]}+1)
Output
对于每组用例,输出满足条件的A序列数量,结果模998244353
Sample Input
2
1 2
2 1
4
1 0 0 2
0 3 0 1
2 1 0 0
0 0 3 1
4
0 1 0 0
1 0 0 0
0 0 0 1
0 0 1 0
Sample Output
Case #1: 6
Case #2: 18
Case #3: 0
Solution
把1~m看作m个点,D矩阵表示这m个点的邻接矩阵,则问题转化为求该图的欧拉回路个数,但是因为求出欧拉回路有重复(在计数时重边看作不同的边)且对于每一条欧拉回路,要选取一个1点把回路切开成为A序列,对BEST’s THEOREM稍作改动有
这里写图片描述
Trees用基尔霍夫矩阵计算即可,时间复杂度O(m^3)
Code

#include<cstdio>#include<algorithm>using namespace std;typedef long long ll;const int maxn=401,maxm=200001,mod=998244353;int T,n,D[maxn][maxn],a[maxn][maxn];int fact[maxm],inv[maxn];void init(){    fact[0]=1;    for(int i=1;i<maxm;i++)fact[i]=(ll)i*fact[i-1]%mod;//求i!    inv[0]=inv[1]=1;    for(int i=2;i<maxn;i++)inv[i]=mod-(int)(mod/i*(ll)inv[mod%i]%mod);//线性求1~n逆元    for(int i=1;i<maxn;i++)inv[i]=(ll)inv[i-1]*inv[i]%mod;//预处理i!的逆元}int inc(int x,int y){    return x+y>=mod?x+y-mod:x+y;}int dec(int x,int y){    return x-y<0?x-y+mod:x-y;}int mod_pow(int a,int b){    int ans=1;    while(b)    {        if(b&1)ans=(ll)ans*a%mod;        a=(ll)a*a%mod;        b>>=1;    }    return ans;}int determinant(int n){    int ans=1;    for(int k=1;k<=n;k++)    {        int pos=-1;        for(int i=k;i<=n;i++)            if(a[i][k])            {                pos=i;                break;            }        if(pos==-1)return 0;        if(pos!=k)            for(int j=k;j<=n;j++)swap(a[pos][j],a[k][j]);        int Inv=mod_pow(a[k][k],mod-2);        for(int i=k+1;i<=n;i++)            if(a[i][k])            {                ans=(ll)ans*Inv%mod;                for(int j=k+1;j<=n;j++)                    a[i][j]=dec(((ll)a[i][j]*a[k][k]%mod),((ll)a[k][j]*a[i][k]%mod));                a[i][k]=0;            }    }    for(int i=1;i<=n;i++)ans=(ll)ans*a[i][i]%mod;    return ans;}int Solve(){    int ans=1;    for(int i=1;i<=n;i++)    {        int in=0,out=0;        for(int j=1;j<=n;j++)in+=D[j][i],out+=D[i][j];        if(in!=out)return 0;        if(i==1)ans=(ll)ans*fact[in]%mod;        else ans=(ll)ans*fact[in-1]%mod;        a[i][i]=in;        for(int j=1;j<=n;j++)a[i][j]=dec(a[i][j],D[i][j]),ans=(ll)ans*inv[D[i][j]]%mod;    }    ans=(ll)ans*determinant(n-1)%mod;    return ans;}int main(){    init();    int Case=1;    while(~scanf("%d",&n))    {        for(int i=1;i<=n;i++)            for(int j=1;j<=n;j++)                scanf("%d",&D[i][j]);        printf("Case #%d: %d\n",Case++,Solve());    }     return 0;}
原创粉丝点击