spfa——洛谷P1608 路徑統計

https://www.luogu.org/problem/show?pid=1608#sub
spfa最短路計數;
我們在spfa裏面修改就好了;
我們增加一個cnt數組,表示到達這個點的最短路總方案;
但是這樣的話在傳遞方面有問題;
我們怎麼辦呢;
對於一個點x;
我們枚舉另一個點k;

if(d[x]+g[x][k]==d[k])cnt[k]+=cnt[x];
    if(d[x]+g[x][k]<d[k]){
        d[k]=d[x]+g[x][k];
        cnt[k]=cnt[x];
    }

關鍵來了;
我們最好要把cnt[x]清空;
這樣的話,cnt數組的值一旦把貢獻全部算好就清空,不會產生重複;

#include<bits/stdc++.h>
#define Ll long long
using namespace std;
const int N=2e3+5;
int g[N][N],d[N],cnt[N];
bool in[N];
int n,m,x,y,z;
void SPFA(int S,int E){
    memset(d,63,sizeof d);
    queue<int>Q;
    Q.push(S);d[S]=0;cnt[S]=1;
    while(!Q.empty()){
        int x=Q.front();Q.pop();in[x]=0;
        if(x==E)continue;
        for(int k=1;k<=n;k++){
            if(d[x]+g[x][k]==d[k])cnt[k]+=cnt[x];
            if(d[x]+g[x][k]<d[k]){
                d[k]=d[x]+g[x][k];
                cnt[k]=cnt[x];
            }
            if(cnt[k]&&!in[k])in[k]=1,Q.push(k);
        }cnt[x]=0;
    }
}
int main()
{
    memset(g,63,sizeof g);
    scanf("%d%d",&n,&m);
    while(m--){
        scanf("%d%d%d",&x,&y,&z);
        g[x][y]=min(g[x][y],z);
    }
    SPFA(1,n);
    if(d[n]==g[0][0])printf("No answer");else printf("%d %d",d[n],cnt[n]);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章