P4336 [SHOI2016]黑暗前的幻想乡

P4336 [SHOI2016]黑暗前的幻想乡

矩阵树定理(高斯消元+乘法逆元)+容斥

ans=总方案数 -(公司1未参加方案数 ∪ 公司2未参加方案数 ∪ 公司3未参加方案数 ∪ ...... ∪ 公司n未参加方案数)

方案数=生成树方案数 所以用矩阵树定理瞎搞

显然后面的部分可以用容斥原理求解

枚举的时候用一个数转成二进制来表示哪些公司参加/不参加

mod=1e9+7是质数所以可以在高斯消元的时候用逆元

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cctype>
using namespace std;
typedef long long ll;
template <typename T> inline void read(T &x){
    char c=getchar(); x=0;
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) x=(x<<3)+(x<<1)+(c^48),c=getchar();
}
const int mod=1e9+7;
inline ll ksm(ll x,int y){
    ll res=1;
    for(;y;y>>=1){
        if(y&1) res=res*x%mod;
        x=x*x%mod;
    }return res;
}
struct edge{int x,y;}a[19][402];
ll f[19][19];
int n,b[19];
inline ll det(){ //高斯消元
    ll res=1; int c=1;
    for(int i=1;i<n;++i){
        int p=i; 
        for(int j=i+1;j<n;++j) if(f[j][i]>f[p][i]) p=j;
        if(p!=i) swap(f[i],f[p]),c=-c; //注意行列式每次交换行符号都会改变
        for(int j=i+1;j<n;++j){
            ll div=f[j][i]*ksm(f[i][i],mod-2)%mod; //除法转成乘逆元
            for(int k=i;k<n;++k) f[j][k]=(f[j][k]-f[i][k]*div%mod+mod)%mod;
        }
        res=res*f[i][i]%mod;
    }return (res*c+mod)%mod;
}
int main(){
    read(n); ll ans=0;
    for(int i=1;i<n;++i){
        read(b[i]);
        for(int j=1;j<=b[i];++j) read(a[i][j].x),read(a[i][j].y);
    }
    for(int i=1;i<=(1<<(n-1))-1;++i){ //i转为二进制数表示方案:0/1 表示是否参加
        memset(f,0,sizeof(f)); //清空重建图
        int p=i,cnt=0;
        for(int j=1;p;++j,p>>=1){ 
            if(!(p&1)) continue; //二进制表示下该位为0->没参加该方案
            for(int k=1;k<=b[j];++k){ //kirchhoff矩阵=度数矩阵-邻接矩阵
                int X=a[j][k].x,Y=a[j][k].y;
                ++f[X][X]; ++f[Y][Y];
                f[X][Y]=(f[X][Y]-1+mod)%mod; //注意减法要重新取模
                f[Y][X]=(f[Y][X]-1+mod)%mod;
            }++cnt;
        }
        ans=(ans+((n-1-cnt)&1 ? -det():det())+mod)%mod; //容斥原理决定是加还是减
    }printf("%lld",ans);
    return 0;
}
原文地址:https://www.cnblogs.com/kafuuchino/p/9669970.html