description
有个点,条边,种颜色,每种颜色都有一个能涂的边集。问构成的满足每个边颜色不同的生成树有多少种。
solution
计数题,有两个限制:
1.生成树方案
2.恰好每种颜色都算入(所有边颜色不同)
1显然能用矩阵树定理。2有待商榷
把所有边算入,求矩阵树行列式得到的是至多种颜色的边。
看到这里就会想到容斥,得到恰好种颜色的边。
code
点击查看代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=18;
const int M=N*N;
const int mod=1e9+7;
struct edge{int x,y;};
vector<edge> V[N];
int n;
ll G[N][N];
ll ksm(ll a,ll b) {ll mul=1;for(;b;b>>=1,a=a*a%mod)if(b&1)mul=mul*a%mod;return mul;}
ll _det() {
// printf("!");
ll mul=1;
for(int i=1;i<=n;i++) {
int r=i;
while(r<=n&&!G[r][i]) r++;
if(r==n+1)continue;
if(r!=i) {mul*=-1;for(int j=1;j<=n;j++) swap(G[r][j],G[i][j]);}
for(int j=i+1;j<=n;j++) {
ll tmp=G[j][i]*ksm(G[i][i],mod-2)%mod;
for(int k=1;k<=n;k++) G[j][k]=(G[j][k]-G[i][k]*tmp)%mod;
}
}
for(int i=1;i<=n;i++)mul=mul*G[i][i]%mod;
// printf("%lld\n",(mul+mod)%mod);
return mul;
}
void init() {
scanf("%d",&n);n--;
for(int i=0;i<n;i++) {
int m;scanf("%d",&m);
for(int j=1;j<=m;j++) {
int u,v;scanf("%d%d",&u,&v);
V[i].push_back((edge){u,v});
}
}
}
void Add(int u,int v) {G[u][u]++;G[v][v]++;G[u][v]--;G[v][u]--;}
void solve() {
int up=1<<n;
ll ans=0;
for(int s=0;s<up;s++) {
// printf("%d:~~~~~~~~~~~~~\n",s);
int bit=0;
for(int j=0;j<n;j++) {
if((1<<j)&s) {
bit++;
for(int k=0;k<V[j].size();k++) {
Add(V[j][k].x,V[j][k].y);
}
}
}
ans=(ans+(((n-bit)&1)?-1:1)*_det())%mod;
for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)G[i][j]=0;
}
printf("%lld",(ans+mod)%mod);
}
int main() {
init();
solve();
return 0;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人