1. 程式人生 > >G - Walk Through the Forest (UVA - 10917)

G - Walk Through the Forest (UVA - 10917)

道路 names urn break blog pri res bsp bre

- 題目大意

一個人,他只會沿著如下條件的道路(A,B)走:存在一條從B出發回家的路徑,比所有從A出發回家的路徑都要短。我們的任務是要找出一共有有多少條不同的回家路徑。

- 解題思路

先用dijkstra預處理出終點到每個點的最短路,然後將滿足行走條件的A、B(除行走條件外,還要滿足一個前提,即A、B之間要有邊)用一條有向邊連起來(A->B),然後利用記憶化搜索來解決這個問題。

- 代碼

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int M=1e5;
const int INF=0x3f3f3f;
int d[1050],vis[1050],g[1050][1050],maps[1050];

void init(int n)
{
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
                g[i][j]=INF;
        }
    }
}
void dijkstra(int s, int n) {
    memset(d, INF, sizeof(d));
    memset(vis, 0, sizeof(vis));
    d[s] = 0;
    while(1) {
        int v = -1;
        for(int u = 1; u <= n; u++) {
            if(!vis[u] && (v == -1 || d[u] < d[v]))
                v = u;
        }
        if(v == -1) break;
        vis[v] = 1;
        for(int u = 1; u <= n; u++) {
            if(!vis[u] && d[u] > d[v] + g[v][u]) {
                d[u] = d[v] + g[v][u];
            }
        }
    }
}

int dfs(int i,int n)
{
    int sum=0;
    if(i==2)
        return 1;
    if(~maps[i])
        return maps[i];
    for(int j=1;j<=n;j++)
    {
        if(g[j][i]<INF&&d[i]>d[j])
            sum+=dfs(j,n);
    }
    maps[i]=sum;
    return maps[i];
}

int main()
{
    int n,m,a,b,c;
    while(scanf("%d",&n)&&n!=0)
    {
        scanf("%d",&m);
        init(n);
       for(int i=1;i<=m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            g[a][b]=c;
            g[b][a]=c;
        }
      memset(maps,-1,sizeof(maps));
       dijkstra(2,n);
       printf("%d\n",dfs(1,n));
    }
    return 0;
}

  

G - Walk Through the Forest (UVA - 10917)