[CSU 1804 有向无环图] DP+拓扑排序

最新推荐文章于 2020-12-20 11:03:31 发布

Xingw-Xiong

最新推荐文章于 2020-12-20 11:03:31 发布

阅读量1.1k

点赞数 1

分类专栏： ACM____数据结构 ACM____动态规划文章标签： dp 拓扑排序

本文链接：https://blog.csdn.net/ACMore_Xiong/article/details/52443159

版权

ACM____数据结构同时被 2 个专栏收录

87 篇文章 0 订阅

订阅专栏

ACM____动态规划

39 篇文章 0 订阅

订阅专栏

[CSU 1804 有向无环图] DP+拓扑排序

题目链接：[CSU 1804 有向无环图]
题目描述：Bobo 有一个 $n$ 个点， $m$ 条边的有向无环图（即对于任意点 $v$ ，不存在从点 $v$ 开始、点 $v$ 结束的路径）。
为了方便，点用 $1,2,\ldots,n$ 编号。设 $count(x,y)$ 表示点 $x$ 到点 $y$ 不同的路径数量（规定 $count(x,x)=0$ ），Bobo想知道

\sum i = 1 n \sum j = 1 n c o u n t (i, j) * a i * b j

$\sum^{n}_{i=1}\sum^{n}_{j=1}count(i,j)*a_i*b_j$ 除以

(109+7) $(10^9+7)$ 的余数。
其中，

ai $a_i$ ,

bj $b_j$ 是给定的数列。
解题思路：这个题目想清楚了真的是一个大水题。但是还是很有灵活性的。可惜省赛的时候脑子不好使。
细细观察上面的式子，对于图中每个点（假设该点为

t $t$ ）,设为

p1,p2,…,pk $p_1,p_2,\ldots,p_k$ 为点

t $t$ 的直接前驱点，我们可以用

cntt $cnt_t$ 表示

∑i=ni=1count(i,t)∗bi $\sum^{i=n}_{i=1}count(i,t)*b_i$ ，显然，

cntt $cnt_t$ 可以由点

t $t$ 的所有直接前驱节点的结果得到。可以写出状态转移方程如下：

c n t t = \sum i = 1 k (c n t p i + a p i)

$cnt_t = \sum^{k}_{i=1}(cnt_{p_i}+a_{p_i})$ 最后,对于所有的点

i(1≤i≤n) $i(1\le i\le n)$ ，乘上对应的

bi $b_i$ ，然后求和就得到答案了。

#include <queue>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

typedef long long LL;

const int MAXN = 1e5 + 5;
const int MOD = 1e9 + 7;

struct Edge {
    int v, next;
    Edge() {}
    Edge(int v, int next) : v(v), next(next) {}
} edges[MAXN];
int N, M;
int head[MAXN], ESZ, IN[MAXN];
LL A[MAXN], B[MAXN], cnt[MAXN];
queue<int> Q;

void init() {
    ESZ = 0;
    memset(head, -1, sizeof(head));
    memset(IN, 0, sizeof(IN));
    memset(cnt, 0, sizeof(cnt));
}
void add_edge(int u, int v) {
    edges[ESZ] = Edge(v, head[u]);
    head[u] = ESZ++;
}
void topu() {
    int u, v;
    while(!Q.empty()) Q.pop();
    for (u = 1; u <= N; u++) {
        if(!IN[u]) Q.push(u);
    }
    while(!Q.empty()) {
        u = Q.front();
        Q.pop();
        for (int i = head[u]; ~i; i = edges[i].next) {
            v = edges[i].v;
            cnt[v] = (cnt[v] + cnt[u] + A[u]) % MOD;
            IN[v] --;
            if(!IN[v]) Q.push(v);
        }
    }
}
int main() {
//    freopen("input.txt", "r", stdin);
    int u, v;
    LL ans;
    while(~scanf("%d %d", &N, &M)) {
        init();
        for(int i = 1; i <= N; i++) {
            scanf("%lld %lld", &A[i], &B[i]);
        }
        for(int i = 1; i <= M; i++) {
            scanf("%d %d", &u, &v);
            add_edge(u, v);
            IN[v] ++;
        }
        topu();
        ans = 0;
        for (int i = 1; i <= N; i++) {
            ans = (ans + cnt[i] * B[i] % MOD) % MOD;
        }
        printf("%lld\n", ans);
    }
    return 0;
}