UVa 10917 A Walk Through the Forest

A Walk Through the Forest

Time Limit:1000MS  Memory Limit:65536K
Total Submit:48 Accepted:15

Description

Jimmy experiences a lot of stress at work these days, especially since his
accident made working difficult. To relax after a hard day, he likes to walk
home. To make things even nicer, his office is on one side of a forest, and his
house is on the other. A nice walk through the forest, seeing the birds and
chipmunks is quite enjoyable.

The forest is beautiful, and Jimmy wants to take a different route everyday. He
also wants to get home before dark, so he always takes a path to make progress
towards his house. He considers taking a path from A to B to be progress if
there exists a route from B to his home that is shorter than any possible route
from A. Calculate how many different routes through the forest Jimmy might
take.

Input

Input contains several test cases followed by a line containing 0. Jimmy
has numbered each intersection or joining of paths starting with 1. His office
is numbered 1, and his house is numbered 2. The first line of each test case
gives the number of intersections N, 1 < N ≤ 1000, and the number of paths
M. The following M lines each contain a pair of intersections a b and an
integer distance 1 ≤ d ≤ 1000000 indicating a path of length d between
intersection a and a different intersection b. Jimmy may walk a path any
direction he chooses. There is at most one path between any pair of
intersections.

Output

For each test case, output a single integer indicating the number of
different routes through the forest. You may assume that this number does not
exceed 2147483647.

Sample
Input

5 6

1 3 2

1 4 2

3 4 3

1 5 12

4 2 34

5 2 24

7 8

1 3 1

1 4 1

3 7 1

7 4 1

7 5 1

6 7 1

5 2 1

6 2 1

0

Sample
Output

2

4

【思路】

最短路+记忆化搜索。

SPFA预处理出每个点到达home的最短距离d，然后沿着d变小的边记忆化搜索路径条数。

【代码】

 #include<cstdio>
 #include<queue>
 #include<vector>
 #include<cstring>
 using namespace std;

 const int maxn = +;
 const int INF=<<;
 struct Edge{
     int u,v,w,next;
 }e[*maxn*maxn];
 int en,front[maxn];

 int n,m;

 inline void AddEdge(int u,int v,int w) {
     en++; e[en].v=v; e[en].w=w; e[en].next=front[u]; front[u]=en;
 }

 int d[maxn];
 void SPFA(int s) {
     int inq[maxn];
     queue<int> q;
     memset(inq,,sizeof(inq));
     for(int i=;i<=n;i++) d[i]=INF;

     d[s]=; inq[s]=; q.push(s);
     while(!q.empty()) {
         int u=q.front(); q.pop(); inq[u]=;
         for(int i=front[u];i>=;i=e[i].next) {
             int v=e[i].v,w=e[i].w;
             if(d[v]>d[u]+w) {
                 d[v]=d[u]+w;
                 if(!inq[v]) {
                     inq[v]=;
                     q.push(v);
                 }
             }
         }
     }
 }

 int f[maxn];
 int dp(int u) {
     int& ans=f[u];
     if(ans>=) return ans;

     if(u==) return ans=;
     ans=;
     for(int i=front[u];i>=;i=e[i].next) {
         int v=e[i].v;
         if(d[v]<d[u]) ans += dp(v);    //只沿着d更小的走
     }
     return ans;
 }
 int main() {
     while(scanf("%d%d",&n,&m)==) {
         en=-;
         memset(front,-,sizeof(front));
         int u,v,w;
         for(int i=;i<m;i++) {
             scanf("%d%d%d",&u,&v,&w);
             AddEdge(u,v,w);
             AddEdge(v,u,w);
         }
         SPFA();
         memset(f,-,sizeof(f));
         printf("%d\n",dp());
     }
     return ;
 }