LOJ10064

LOJ10064

首先, Di=DSTi D i = D S T i (PRIM中)
所以可以先求一趟PRIM维护最小路径生成树
再根据乘法原理来统计,连乘到达每个点的答案

#include
#define LL long long
#define gt() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++)
using namespace std;
static char buf[1000000],*p1=buf,*p2=buf;
const int maxn=1005,INF=1e9,TT=2147483647;
int n,m,g[maxn][maxn],dst[maxn],cnt[maxn];bool vis[maxn];LL Ans=1;
int read(){
    int ret=0;char ch=gt();
    while(ch<'0'||ch>'9') ch=gt();
    while(ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=gt();
    return ret;
}
int main(){
    n=read(),m=read();
    for(int i=1;i<=n;i++)
      for(int j=1;j<=n;j++) g[i][j]=INF;
    for(int i=1;i<=m;i++){
        int x=read(),y=read();
        g[x][y]=g[y][x]=read();
    }
    for(int i=2;i<=n;i++) dst[i]=g[1][i];
    for(int i=1;iint min_x=INF,k;
        for(int j=2;j<=n;j++)if(!vis[j]&&dst[j]1;
        for(int j=2;j<=n;j++)if(!vis[j]&&(min_x=dst[k]+g[k][j])for(int i=1;i<=n;i++)
      for(int j=2;j<=n;j++) if(i!=j&&dst[i]+g[i][j]==dst[j]) ++cnt[j];
    for(int i=2;i<=n;i++) (Ans*=cnt[i])%=TT;
    printf("%lld\n",Ans);
    return 0;
}

你可能感兴趣的:(最小生成树,LOJ)