【HDU6026】Deleting Edges

题意

有一个n个节点的无向图，结点编号从0-n-1，每条边的长度时1to9的一个正整数。现在要删除一些边(或者不删)，使得到的新图满足下面两个要求。

1.
新图是一颗树有n-1条边
2.
对于每个结点v（0ton-1）从0到v的距离要等于原图上从0到v的最短路

现在计算有多少种不同的图满足上面两个条件

分析

统计每个点可以到达最短路的入度值，然后相乘。这个题因为spfa写错了调了老久···

 #include <cstdio>

 #include <cstring>

 #include <algorithm>

 #include <iostream>

 #include <queue>

 using namespace std;

 const int INF=;

 const int maxn=;

 const int MOD=;

 int n;

 char g[maxn][maxn];

 int G[maxn][maxn];

 int d[maxn],vis[maxn];

 long long in[maxn];

 long long ans;

 void spfa(){

     memset(vis,,sizeof(vis));

     for(int i=;i<=n;i++)d[i]=INF;

     queue<int>q;

     q.push();vis[]=;

     d[]=;

     while(!q.empty()){

         int u=q.front();q.pop();

         vis[u]=;

         for(int i=;i<n;i++){

             if(G[u][i]){

                 if(d[i]>=d[u]+G[u][i]){

                     d[i]=d[u]+G[u][i];

                     if(!vis[i]){

                       vis[i]=;

                       q.push(i);

                     }

                 }

             }

         }

     }

     return;

 }

 int main(){

     while(scanf("%d",&n)!=EOF){

         memset(G,,sizeof(G));

         for(int i=;i<n;i++){

             for(int j=;j<n;j++){

                 scanf(" %c",&g[i][j]);

             }

         }

         for(int i=;i<n;i++){

             for(int j=;j<n;j++){

                 G[i][j]=g[i][j]-'';

             }

         }

         spfa();

         memset(in,,sizeof(in));

         for(int i=;i<n;i++){

             for(int j=;j<n;j++){

                 if(i!=j){

                     if(G[i][j]&&d[j]==d[i]+G[i][j]){

                         in[j]++;

                     }

                 }

             }

         }

         ans=;

         for(int i=;i<n;i++){

             if(in[i])

             ans=(ans%MOD*in[i]%MOD)%MOD;

         }

         printf("%lld\n",ans);

     }

 return ;

 }