BZOJ 4011 HNOI2015 落忆枫音拓扑序DP

最新推荐文章于 2019-03-03 11:33:00 发布

PoPoQQQ

最新推荐文章于 2019-03-03 11:33:00 发布

阅读量5k

点赞数

CC 4.0 BY-SA版权

分类专栏： BZOJ 拓扑排序动态规划文章标签： BZOJ BZOJ4011 拓扑排序动态规划

本文链接：https://blog.csdn.net/popoqqq/article/details/45194103

BZOJ 同时被 3 个专栏收录

667 篇文章

订阅专栏

动态规划

50 篇文章

订阅专栏

拓扑排序

10 篇文章

订阅专栏

博客探讨了如何解决BZOJ 4011题目的方法，该题目涉及在有向无环图（DAG）中添加边并计算以1为根的树形子图数量。通过朱刘算法的推论，得出原始答案，然后考虑新边引入后不合法方案的排除。采用动态规划（DP）进行计算，定义状态fi表示从某个节点y到节点i的路径上排除中间点的点集贡献，并给出动规方程和初始条件。

题目大意：给定一张有向无环图，现在要求加入一条边，求加入后以1为根的树形图个数
首先不考虑加入的这条边，那么这个图是一个DAG
由朱刘算法的推论可知，如果除根节点外每个点都选择一条入边，由于没有环，因此一定会形成一个树形图
因此答案就是 $\prod_{i=2}^ndegree_i$ 其中 $degree_i$ 表示第 $i$ 个点的入度
现在加入这条边之后，我们仍然可以套用这个公式，但是这样就会有一些不合法的方案被统计进来，我们需要把这些不合法的方案减掉
一个方案如果不合法，那么一定会形成一个环，而环一定包含新加入的那条边
因此我们减掉的方案其实是：
$\sum_{S是原图中y\rightarrow x的一条路径的点集}\prod_{2\leq j\leq n,j\notin S}degree_j$
然后我们就可以DP了
令 $f_i$ 表示 $\sum_{S是原图中y\rightarrow i的一条路径的点集}\prod_{2\leq j\leq n,j\notin S}degree_j$
那么有动规方程
$f_i=\frac{\sum_{j\rightarrow i}f_j}{degree_i}$
初值
$f_y=\frac{\prod_{i=2}^ndegree_i}{degree_y}$
注意 $y=1$ 时要特判= =

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 200200
#define MOD 1000000007
using namespace std;
struct abcd{
    int to,next;
}table[M];
int head[M],tot;
int n,m,s,t;
int degree[M],_degree[M];
long long inv[M],f[M],ans=1;
void Add(int x,int y)
{
    table[++tot].to=y;
    table[tot].next=head[x];
    head[x]=tot;
}
void Linear_Shaker()
{
    int i;
    for(inv[1]=1,i=2;i<=m+1;i++)
        inv[i]=(MOD-MOD/i)*inv[MOD%i]%MOD;
}
void Topology_Sort()
{
    static int q[M];
    int i,r=0,h=0;
    f[t]=ans;
    for(i=1;i<=n;i++)
        if(!degree[i])
            q[++r]=i;
    while(r!=h)
    {
        int x=q[++h];
        (f[x]*=inv[_degree[x]])%=MOD;
        for(i=head[x];i;i=table[i].next)
        {
            (f[table[i].to]+=f[x])%=MOD;
            if(!--degree[table[i].to])
                q[++r]=table[i].to;
        }
    }
}
int main()
{
    int i,x,y;
    cin>>n>>m>>s>>t;
    Linear_Shaker();
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        Add(x,y);
        degree[y]++;
        _degree[y]++;
    }
    _degree[t]++;
    for(i=2;i<=n;i++)
        (ans*=_degree[i])%=MOD;
    if(t==1)
    {
        cout<<ans<<endl;
        return 0;
    }
    Topology_Sort();
    cout<<(ans-f[s]+MOD)%MOD<<endl;
    return 0;
}