uva 10917 Walk Through the Forest

d[i]表示从i点到家的距离，如果d[i] > d[j[且i,j之间存在路的话，那么就可以走i->j。最后问从起点到终点有几条不同的路径。

首先用dijkstra求出每一点到终点的距离。

然后建立新图，如上所述，如果d[i] > d[j[且i,j之间存在路的话，那么建一条i->j的边。可以证明，建立的新图是一个DAG。

最后在新图上dp。dp[i]表示i到终点有几条路径。dp[i] = 所有它指向的点的dp之和。

#include <iostream>
#include<vector>
#include<stdio.h>
#include<cstring>
#include<queue>
#define maxn 1010
#define INF 2000000000
using namespace std;
int  n,m;
struct Edge{
   int from,to,dist;
};
struct HeapNode{
    int d,u;
    bool operator < (const HeapNode& rhs) const
    {
        return d > rhs.d;
    }
};
vector<Edge>edges;
vector<int>G[maxn];
bool vis[maxn];
int d[maxn]; //最短路距离
int p[maxn]; //存最短路上的上一个节点标号
int map[maxn][maxn]; //存修改后的图
int map1[maxn][maxn]; //存原始图
int dp[maxn];
int ans;

void init()
{
    for(int i = 1;i <= n;i++) G[i].clear();
    edges.clear();
    for(int i = 1;i <= n;i++)
        for(int j = 1;j <= n;j++)
        {
            map[i][j] = -1; //map只需记录两点间有无路径即可，不许记录权值
            map1[i][j] = -1;
        }
}

void addedge(int from,int to,int dist)
{
    edges.push_back((Edge){from,to,dist});
    int mm = edges.size();
    G[from].push_back(mm - 1);
}

void dijkstra(int s) //从家的节点开始计算最短路
{
    priority_queue<HeapNode>Q;
    for(int i = 1;i <= n;i++) d[i] = INF;
    d[2] = 0;
    memset(vis,0,sizeof vis);
    Q.push((HeapNode){0,s});
    while(!Q.empty())
    {
        HeapNode x = Q.top(); Q.pop();
        int u = x.u;
        if(vis[u]) continue;
        vis[u] = 1;
        for(int i = 0;i < G[u].size();i++)
        {
            Edge& e = edges[G[u][i]];
            if(d[e.to] > d[u] + e.dist)
            {
                d[e.to] = d[u] + e.dist;
                p[e.to] = u;
                Q.push((HeapNode){d[e.to],e.to});
            }
        }
    }
}

void create_new() //建立新图
{
    for(int i = 1;i <= n;i++)
        for(int j = 1;j <= n;j++)
        {
            if(i == j) continue;
            else if(d[i] > d[j] && map1[i][j] == 1) map[i][j] = 1;
        }
}

int dag_dp(int cur)
{
    if(dp[cur] != -1) return dp[cur];
    int tmp = 0;
    for(int i = 1;i <= n;i++)
    {
        if(map[cur][i] == 1) tmp+=dag_dp(i);
    }
    return dp[cur] = tmp;
}

int main()
{
    while(scanf("%d",&n),n)
    {
        scanf("%d",&m);
        init();
        for(int i = 1;i <= m;i++)
        {
            int from,to,dist;
            scanf("%d %d %d",&from,&to,&dist);
            addedge(from,to,dist);
            addedge(to,from,dist);
            map1[from][to] = map1[to][from] = 1; 
        }
        dijkstra(2);
        create_new();
        memset(dp,-1,sizeof dp);
        dp[2] = 1; //从2节点到2节点有一条路
        ans = dag_dp(1);
        printf("%d\n",ans);
    }
    return 0;
}