UVA - 10917 Walk Through the Forest (最短路+DP)

標籤：style   os   io   for   ar   line   amp   size   

題意：Jimmy打算每天沿著一條不同的路走，而且，他只能沿著滿足如下條件的道路（A，B）：存在一條從B出發回家的路徑，比所有從A出發回家的路徑都短，你的任務是計算有多少條不同的路徑

從後往前找最短路，

對於每一步要更新之後走的位置值；

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int INF=10000000;
const int MAXN=1024;
int dp[MAXN],dist[MAXN],cost[MAXN][MAXN];
int m,n;
int s,e,w;
int vis[MAXN];
void Dij()
{
    int Min,cnt;
    for(int i=1;i<=n;i++)
    {
        dist[i]=cost[2][i];
    }
    vis[2]=1;
    for(int i=1;i<=n;i++)
    {
        Min=INF;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&dist[j]<Min)
            {
                Min=dist[j];
                cnt=j;
            }
        }
        vis[cnt]=1;
        if(Min==INF)
            break;
        for(int k=1;k<=n;i++)
            if(!vis[k]&&dist[k]>dist[cnt]+cost[k][cnt])
              dist[k]=dist[cnt]+cost[k][cnt];//更新之後每個與cnt有聯絡的點
    }
}
int dfs(int cur)
{
    if(dp[cur]>0)
        return dp[cur];
    int ans=0;
    for(int i=1;i<=n;i++)
        if(dist[cur]>dist[i]&&cost[cur][i]!=INF)
              ans+=dfs(i);
    return dp[cur]=ans;
}
int main()
{
    while(~scanf("%d",&n))
    {
        if(n==0)
            break;
        scanf("%d",&m);
        memset(cost,INF,sizeof(cost));
        memset(dist,INF,sizeof(dist));
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++)
            cost[i][i]=0;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&s,&e,&w);
            cost[s][e]=w;cost[e][s]=w;
        }
        Dij();
        dp[2]=1;
        int ans=dfs(1);//深搜找
        printf("%d\n",ans);
    }
    return 0;
}