求DAG上两点的最短距离

Problem

给出一个不带边权(即边权为1)的有向无环图(unweighted DAG)以及DAG上两点s, t,求s到t的最短距离,如果无法从s走到t,则输出-1。

Solution

DFS,BFS都可,对于unweighted DAG, BFS更合适,下面给出DFS解法。

const int N(1e5+5);
vector<int> g[N];
int d[N], vis[N];

void dfs(int u, int t){
    vis[u]=1; if(u==t){d[u]=0; return;}
    for(int i=0; i<g[u].size(); i++){
        int &v=g[u][i]; if(!vis[v]) dfs(v, t);
        if(~d[v]) d[u]=~d[u]?min(d[v]+1, d[u]):d[v]+1;
    }
}

int solve(int s, int t){
    memset(d, -1, sizeof(d));
    memset(vis, 0, sizeof(vis));
    dfs(s, t);
    return d[s];
}

对于weighted DAG, 解法类似。

 

posted @ 2015-09-19 22:47  Pat  阅读(369)  评论(0编辑  收藏  举报