【bzoj4596】[Shoi2016]黑暗前的幻想乡 容斥原理+矩阵树定理

题目描述

给出 n 个点和 n1 种颜色,每种颜色有若干条边。求这张图多少棵每种颜色的边都出现过的生成树,答案对 109+7 取模。

输入

第一行包含一个正整数 N(N<=17), 表示城市个数。
接下来 N-1 行,其中第 i行表示第 i个建筑公司可以修建的路的列表:
以一个非负数mi 开头,表示其可以修建 mi 条路,接下来有mi 对数,
每对数表示一条边的两个端点。其中不会出现重复的边,也不会出现自环。

输出

仅一行一个整数,表示所有可能的方案数对 10^9 + 7 取模的结果。

样例输入

4
2 3 2 4 2
5 2 1 3 1 3 2 4 1 4 3
4 2 1 3 2 4 1 4 2

样例输出

17


题解

容斥原理+矩阵树定理

答案为:随便选的 - 钦定1种颜色不能选的 + 钦定2种颜色不能选的 - ... 。

爆搜每种颜色是否被钦定不能选,然后使用矩阵树定理求出当前条件下的生成树个数即可。

时间复杂度 O(2n·n3)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
#include <cstdio>
#include <cstring>
#include <algorithm>
#define mod 1000000007
typedef long long ll;
using namespace std;
int n , m[18] , vx[18][140] , vy[18][140] , v[18];
ll a[18][18] , ans;
inline ll pow(ll x , int y)
{
    ll ans = 1;
    while(y)
    {
        if(y & 1) ans = ans * x % mod;
        x = x * x % mod , y >>= 1;
    }
    return ans;
}
void dfs(int x , int flag)
{
    if(x == n)
    {
        memset(a , 0 , sizeof(a));
        int i , j , k;
        ll now = 1 , t;
        for(i = 1 ; i < n ; i ++ )
            if(v[i])
                for(j = 1 ; j <= m[i] ; j ++ )
                    a[vx[i][j]][vx[i][j]] ++ , a[vy[i][j]][vy[i][j]] ++ , a[vx[i][j]][vy[i][j]] -- , a[vy[i][j]][vx[i][j]] -- ;
        for(i = 2 ; i <= n ; i ++ )
        {
            for(j = i ; j <= n ; j ++ )
                if(a[j][i])
                    break;
            if(j > n) break;
            if(j != i)
            {
                flag = -flag;
                for(k = i ; k <= n ; k ++ )
                    swap(a[i][k] , a[j][k]);
            }
            now = now * a[i][i] % mod , t = pow(a[i][i] , mod - 2);
            for(j = i ; j <= n ; j ++ ) a[i][j] = a[i][j] * t % mod;
            for(j = i + 1 ; j <= n ; j ++ )
                for(t = a[j][i] , k = i ; k <= n ; k ++ )
                    a[j][k] = (a[j][k] - t * a[i][k] % mod + mod) % mod;
        }
        if(i > n) ans = (ans + flag * now + mod) % mod;
        return;
    }
    v[x] = 1 , dfs(x + 1 , flag);
    v[x] = 0 , dfs(x + 1 , -flag);
}
int main()
{
    int i , j;
    scanf("%d" , &n);
    for(i = 1 ; i < n ; i ++ )
    {
        scanf("%d" , &m[i]);
        for(j = 1 ; j <= m[i] ; j ++ )
            scanf("%d%d" , &vx[i][j] , &vy[i][j]);
    }
    dfs(1 , 1);
    printf("%lld\n" , ans);
    return 0;
}

 

 

posted @   GXZlegend  阅读(972)  评论(0编辑  收藏  举报
编辑推荐:
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 基于Microsoft.Extensions.AI核心库实现RAG应用
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 地球OL攻略 —— 某应届生求职总结
· 提示词工程——AI应用必不可少的技术
· Open-Sora 2.0 重磅开源!
· 周边上新:园子的第一款马克杯温暖上架
点击右上角即可分享
微信分享提示