hdu-6714 最短路 2(dijkstra+堆優化)

hdu-6714

小 A 是社團裏的工具人,有一天他的朋友給了他一個 n 個點,m 條邊的正權連通無向圖,要他計算所有點兩兩之間的最短路。

作爲一個工具人,小 A 熟練掌握着 floyd 算法,設 w[i][j] 爲原圖中 (i,j) 之間的權值最小的邊的權值,若沒有邊則 w[i][j]=無窮大。特別地,若 i=j,則 w[i][j]=0。

Floyd 的 C++ 實現如下:

```c++
for(int k=1;k<=p;k++)
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
  w[i][j]=min(w[i][j],w[i][k]+w[k][j]);
```

當 p=n 時,該代碼就是我們所熟知的 floyd,然而小 A 爲了讓代碼跑的更快點,所以想減少 p 的值。

令 Di,j 爲最小的非負整數 x,滿足當 p=x 時,點 i 與點 j 之間的最短路被正確計算了。

現在你需要求 ∑ni=1∑nj=1Di,j,雖然答案不會很大,但爲了顯得本題像個計數題,你還是需要將答案對 998244353 取模後輸出。

輸入:

第一行一個正整數 T(T≤30) 表示數據組數

對於每組數據:

第一行兩個正整數 n,m(1≤n≤1000,m≤2000),表示點數和邊數。

保證最多隻有 5 組數據滿足 max(n,m)>200 

接下來 m 行,每行三個正整數 u,v,w 描述一條邊權爲 w 的邊 (u,v),其中 1≤w≤109

輸出:

輸出 T 行,第 i 行一個非負整數表示第 i 組數據的答案

題意:求最短路上最大鬆弛點的最小值

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <map>
#include <iterator>
#include <vector>
#include <set>
#include <bitset>
#include <stack>
#define mems(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const ll mod=998244353;;
const int N=2e3+10;
const int inf=0x3f3f3f3f;
int head[N],tot,ans[N];
bool vis[N];
ll dis[N];
struct edge
{
    int v,net;
    ll w;
} e[N*2];
struct node
{
    ll w;
    int u;
    bool friend operator <(node a,node b)
    {
        return a.w>b.w;
    }
};
void init()
{
    tot=0;
    mems(head,-1);
}
void add(int u,int v,int w)
{
    e[tot].v=v;
    e[tot].w=w;
    e[tot].net=head[u];
    head[u]=tot++;
}
void dijkstra(int s)
{
    mems(dis,inf);
    mems(ans,0);
    mems(vis,0);
    priority_queue<node>q;
    q.push(node{0,s});
    dis[s]=0;
    while(!q.empty())
    {
        node now=q.top();
        q.pop();
        int u=now.u;
        if(vis[u])
            continue;
        vis[u]=1;
        for(int i=head[u];~i;i=e[i].net)
        {
            int v=e[i].v;
            ll w=e[i].w;
            if(dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                if(u!=s)
                    ans[v]=max(ans[u],u);
                q.push(node{dis[v],v});
            }
            else if(dis[v]==dis[u]+w)//多條最短路,取最小
            {
                if(u!=s)
                    ans[v]=min(ans[v],max(ans[u],u));
            }
        }
    }
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        init();
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1; i<=m; i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
        }
        ll sum=0;
        for(int i=1;i<=n;i++)
        {
            dijkstra(i);
            for(int j=1;j<=n;j++)
                sum=(sum+ans[j])%mod;
        }
        printf("%lld\n",sum);
    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章