2016年湖南省第十二届大学生程序设计大赛 B题 “有向无环图” CSG - 1135（dfs，记忆化搜索，逆元）

最新推荐文章于 2026-06-21 17:05:31 发布

原创最新推荐文章于 2026-06-21 17:05:31 发布 · 232 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#深度优先 #算法

湖南acm省赛同时被 3 个专栏收录

10 篇文章

订阅专栏

图论

7 篇文章

订阅专栏

搜索

5 篇文章

订阅专栏

该博客详细介绍了2016年湖南省第十二届大学生程序设计大赛B题，题目涉及有向无环图（DAG）上的路径计数问题。博主首先阐述了题目的内容，要求计算每个节点到每个节点的路径数量，乘以起点的a值和终点的b值，并模1e9+7。接着，博主提供了样例解释和解题思路，强调从入度为零的节点开始使用深度优先搜索（DFS），利用记忆化搜索优化算法，通过计算逆元来处理路径数的除法操作。最后，博客中包含了解题代码和小结，总结了整个DFS过程的重要性。

题目链接

一.题目内容

给一个有向无环图，求图中每个点到每个点的路径条数 * 起点的a值 * 终点的b值，结果模1e9+7

样例解释

样例一：
1到2，一条路径，1a1b2=1，1到3，两条路径，2a1b3=2，2到3，一条路径，1a2b3=1，结果为4

二.解题思路

最开始的起点肯定是入度为零的点，从它们开始深搜。

若儿子为叶子节点，则直接等于一条新路径，如下图 k1/k2/k3为叶子，则sumj的值为aj * (bk1,bk2…)
而 i 为 j 的父亲，所以 j 所能到达的点 i 也能到达，并且路径数是相同的，路径为 i->j->xxx，所以 i 对这些点的值等于 j 对它们的值 / aj 再 * ai，并且 i->j 也是一条新路径，所以还要加上 i 对 j 的值 ai * bj。

因为要对1e9+7取模，所以 / aj 这一步可用费马小定理 * aj的逆元即可。

因为可能有多次搜索，所以用一个数组存点对它所能到达的点的值，实现记忆化搜索，不然肯定会超时。
在这里插入图片描述

三.解题代码

#include<bits/stdc++.h>
using namespace std;
#define rep(i,l,r) for(int i=l;i<=r;++i)
#define rrep(i,l,r) for(int i=l;i>=r;--i)
#define int long long
int dir[4][2]= {{-1,0},{0,-1},{1,0},{0,1}};
const int N=1e6+10,M=1e9+7;
//-----------//
vector<int>v[N];
int a[N],b[N],d[N],sum[N];
int fp(int b,int p) {
    int res=1;
    while(p) {
        if(p&1) res=res*b%M;
        b=b*b%M;
        p>>=1;
    }
    return res;
}
int inv(int x) {
    return fp(x,M-2);
}
int dfs(int x) {
    if(sum[x]) return sum[x];//若搜索过了直接返回值
    int res=0;
    for(auto i:v[x]) {
        res=(res+a[x]*b[i]%M)%M; //x->i的新路径值
        int temp=dfs(i)*inv(a[i])%M; // x->（i的子孙）的路径值
        res=(res+a[x]*temp%M)%M;
    }
    return sum[x]=res;//记忆化
}
signed main() {
    sync;
    int n,m;
    while(cin>>n>>m) {
        rep(i,1,n) {
            cin>>a[i]>>b[i];
            v[i].clear();
            d[i]=sum[i]=0;
        }
        while(m--) {
            int x,y;
            cin>>x>>y;
            v[x].push_back(y);
            d[y]++;
        }
        int ans=0;
        rep(i,1,n) {
            if(!d[i]&&v[i].size()) dfs(i);
        }
        rep(i,1,n) ans=(ans+sum[i])%M;
        cout<<ans<<endl;
    }
}

四.小结

弄清dfs过程后，做法就显而易见了