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]);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章