hdu 1142 A Walk Through the Forest

http://acm.hdu.edu.cn/showproblem.php?pid=1142

这道题是spfa求最短路，然后dfs()求路径数。

 #include <cstdio>
 #include <queue>
 #include <cstring>
 #include <algorithm>
 #define maxn 1001
 using namespace std;
 const int inf=<<;

 int g[maxn][maxn];
 int dis[maxn];
 bool vis[maxn],visi[maxn];
 int n,m,a,b,d;
 int di;
 int cnt[maxn];
 int num[maxn];

 bool spfa()
 {
     queue<int>q;
     memset(vis,false,sizeof(vis));
     memset(cnt,,sizeof(cnt));
     for(int i=; i<=n; i++) dis[i]=inf;
     dis[]=;
     vis[]=true;
     q.push();
     while(!q.empty())
     {
         int u=q.front();
         q.pop();
         vis[u]=false;
         for(int i=; i<=n; i++)
         {
             if(dis[i]>dis[u]+g[u][i]&&g[u][i]!=inf)
             {
                 dis[i]=dis[u]+g[u][i];
                 if((++cnt[i])>n) return false;
                 if(!vis[i])
                 {
                     q.push(i);
                     vis[i]=true;
                 }
             }
         }
     }
     return true;
 }

 void dfs(int src)
 {
     if(src==)
     {
         num[]=;
         return ;
     }
     int sum=;
     for(int i=; i<=n; i++)
     {
         if(g[src][i]!=inf&&dis[src]>dis[i])
         {
             if(num[i]>=)
             {
                 sum+=num[i];
             }
             else
             {
                 dfs(i);
                 sum+=num[i];
             }
         }
     }
     num[src]=sum;
 }
 int main()
 {
     while(scanf("%d",&n)!=EOF)
     {
         if(n==) break;
         scanf("%d",&m);
         for(int i=; i<=n; i++)
         {
             for(int j=; j<=n; j++)
             {
                 if(i==j) g[i][j]=;
                 else g[i][j]=inf;
             }
         }
         for(int i=; i<m; i++)
         {
             scanf("%d%d%d",&a,&b,&d);
             g[a][b]=g[b][a]=min(g[a][b],d);
         }
         spfa();
         //printf("%d\n",dis[1]);
         for(int i=; i<=n; i++)
         {
             num[i]=-;
         }
         dfs();
         printf("%d\n",num[]);
     }
     return ;
 }