HDU-6714 最短路 2 （最短路Dijstra+Floyd理解）

鏈接：http://acm.hdu.edu.cn/showproblem.php?pid=6714

題意：就是求任意兩點最短路中鬆弛點中最小值的和。

思路：Floyd第一層循環其實就是枚舉的鬆弛點。枚舉起點，每次跑一遍Dijstra，求所有鬆弛點的最小值即可。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 1e3+10;
const int M = 4e3+10;
const ll mod = 998244353 ;
struct node
{
    int to,nxt;
    ll w;
}g[M];
struct Node
{
    int to;
    ll w;
    Node(){}
    Node(int to,ll w):to(to),w(w){} 
    friend bool operator<(Node a,Node b)
    {
		return a.w>b.w;
    }
}now;
int head[N],cnt;
priority_queue <Node> q;
int vis[N];
ll dis[N],ans;
int D[N][N]; 
int n,m;
void Init()
{
    ans=cnt=0;
    for(int i=1;i<=n;i++)
        head[i]=-1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
                D[i][j]=0;             
}
void add(int u,int v,ll w)
{
    g[cnt].to=v; g[cnt].w=w; g[cnt].nxt=head[u]; head[u]=cnt++;
}
void dijs(int s)
{
    for(int i=1;i<=n;i++)
        dis[i]=1e18,vis[i]=0;    
    int u,v;
    dis[s]=0;
    q.push(Node(s,0));
    while(!q.empty())
    {
        now=q.top();
        q.pop();
        u=now.to;
        if(vis[u]) continue;
        vis[u]=1;
        for(int i=head[u];~i;i=g[i].nxt)
        {
            v=g[i].to;
            if(dis[v]>dis[u]+g[i].w)
            {
                dis[v]=dis[u]+g[i].w;
                q.push(Node(v,dis[v]));
                if(dis[v]==g[i].w) D[s][v]=0;
                else  D[s][v]=max(D[s][u],u);
            }
            else if(dis[v]==dis[u]+g[i].w)
                D[s][v]=min(D[s][v],max(D[s][u],u));
        }
    }
}
int main(void)
{
    int u,v;
    ll w;
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        Init();
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d%lld",&u,&v,&w);
            add(u,v,w),add(v,u,w);
        }
        for(int i=1;i<=n;i++)
            dijs(i);    
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
				ans=(ans+D[i][j])%mod;
        printf("%lld\n",ans);    
    }
    return 0;
}