Deleting Edges 【HDU - 6026】

题目链接

题目大意

现在有一个带权图，要你删去一些边（也可以不删）使得剩下的边和点组成的图满足：
1.有n个点n-1条边。
2.现在的这个图所有的点到0点的距离等于原图上这个点到0点的最短路径
问你有多少种删法。

解题思路

最短路+思路
首先我们用最短路算法算出在原图中0点到每个点的最短路径
然后如果有一对点满足dis[j] == dis[i] + e[i][j]那么说明这条边是最短路上的一条边，所以保存一下，相乘就好

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
char s[55];
int e[105][105],a[105],dis[105],book[105];
int inf=0x3f3f3f3f;
long long mod=1e9+7;
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        memset(a,0,sizeof(a));
        for(int i=1;i<=n;i++)
        {
            scanf("%s",s+1);
            for(int j=1;j<=n;j++)
            {
                if(i==j)
                    e[i][j]=0;
                else if(s[j]=='0')
                    e[i][j]=inf;
                else
                    e[i][j]=s[j]-'0';
            }
        }
        for(int i=1;i<=n;i++)
            dis[i]=e[1][i];
        memset(book,0,sizeof(book));
        book[1]=1;
        int u;
        for(int i=1;i<=n-1;i++)
        {
            int minn=inf;
            for(int j=1;j<=n;j++)
            {
                if(book[j]==0&&dis[j]<minn)
                {
                    minn=dis[j];
                    u=j;
                }
            }
            book[u]=1;
            for(int v=1;v<=n;v++)
            {
                if(e[u][v]<inf&&dis[v]>dis[u]+e[u][v]&&book[v]==0)
                    dis[v]=dis[u]+e[u][v];
            }
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                if(e[i][j]<inf&&dis[j]==dis[i]+e[i][j]&&i!=j)
                    a[j]++;
            }
        }
        long long ans=1;
        for(int i=1;i<=n;i++)
        {
            if(a[i])
                ans=((long long)(ans*a[i]))%mod;
        }
        printf("%lld\n",ans);
    }
    return 0;
}