洛谷P3953 [NOIP2017] 逛公园（最短路+拓扑序+DP）

洛谷P3953 [NOIP2017] 逛公园

原题地址：https://www.luogu.org/problemnew/show/P3953

题意：
策策同学特别喜欢逛公园。公园可以看成一张NN个点MM条边构成的有向图，且没有 自环和重边。其中1号点是公园的入口，NN号点是公园的出口，每条边有一个非负权值， 代表策策经过这条边所要花的时间。

策策每天都会去逛公园，他总是从1号点进去，从NN号点出来。

策策喜欢新鲜的事物，它不希望有两天逛公园的路线完全一样，同时策策还是一个 特别热爱学习的好孩子，它不希望每天在逛公园这件事上花费太多的时间。如果1号点 到NN号点的最短路长为dd，那么策策只会喜欢长度不超过d + Kd+K的路线。

策策同学想知道总共有多少条满足条件的路线，你能帮帮它吗？

为避免输出过大，答案对PP取模。

如果有无穷多条合法的路线，请输出−1。

数据范围
$1 \le P \le 10^9,1 \le a_i,b_i \le N ,0 \le c_i \le 10001≤P≤10$

题解：
当时考场上居然什么思路都没有，看来是被T2大模拟搞傻了。

首先，与最短路条数的DP相似的想法：
dp[i][j]表示到点i，长度与最短路相差j的路径条数。
如果没有环，转移就可以直接按拓扑序。

那么有环怎么办？
既然我们已经有了dp[i][j]这个思路，那么顺势想过来，
如果把dp[i][j]看成n*k个点，只要没有0环，他们的转移就是没有环的。
于是对这n*k个点重新跑拓扑序DP即可。
如果有点没有被拓扑到，说明存在0环，输出-1。

这题卡常小心。

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define id(x , y) (x + (y) * n)
using namespace std;
const int N=100010;
const int M=200010;
const int inf=0x3f3f3f3f;
struct node
{
    int x,y;
    node(){}
    node(int x,int y):x(x),y(y){}
    bool friend operator<(const node &A,const node &B){return A.y>B.y;}
};
priority_queue<pair<int,int> > P;
int T=0,head[N],to[M],nxt[M],w[M],num=0,n,m,k,mod,dis[N],cnt=0,dp[N*51],du[N*51],h[N*51],t[N*51],q[M*51],Q[N*51];
bool vis[N];
inline int read()
{
    int ret=0,w=1; char ch=getchar();
    while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar(); if(ch=='-'){w=-1; ch=getchar();}
    while(ch>='0'&&ch<='9') {ret=(ret<<1)+(ret<<3)+ch-'0'; ch=getchar();}
    return ret*w;
}
void build(int u,int v,int ww) {num++; to[num]=v; w[num]=ww; nxt[num]=head[u]; head[u]=num;}
void dijkstra()
{
    memset(dis,0x3f,sizeof(dis));
    dis[1]=0; P.push(make_pair(0,1));
    while(!P.empty())
    {
        int u=P.top().second;P.pop();
        if(vis[u]) continue; vis[u]=1;
        for(int i=head[u];i;i=nxt[i])
        {
            int v=to[i];
            if(dis[v]>dis[u]+w[i])
            {
                dis[v]=dis[u]+w[i];
                P.push(make_pair(-dis[v],v));
            }
        }
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        n=read(); m=read(); k=read(); mod=read();
        memset(head,0,sizeof(head)); memset(vis,0,sizeof(vis)); memset(du,0,sizeof(du)); memset(dp,0,sizeof(dp)); num=0; cnt=0;
        register int  i,j,u,v,ww,lf=1,rg=0;
        for(i=1;i<=m;i++)
        {u=read(); v=read(); ww=read(); build(u,v,ww);}
        dijkstra();
        for(u=1;u<=n;u++)
        {
            for(j=0;j<=k;j++)
            {
                h[id(u,j)]=cnt;
                for(i=head[u];i;i=nxt[i])
                {
                    int v=to[i];
                    if(dis[u]+w[i]+j<=dis[v]+k)
                    {
                        du[id(v,dis[u]+w[i]+j-dis[v])]++;
                        q[++cnt]=id(v,dis[u]+w[i]+j-dis[v]);
                    }
                }
                t[id(u,j)]=cnt;
            }
        }   
        for(i=1;i<=id(n,k);i++) if(du[i]==0) Q[++rg]=i;
        dp[id(1,0)]=1;
        while(lf<=rg)
        {       
            int x=Q[lf]; lf++;
            for(i=h[x]+1;i<=t[x];i++)
            {
                du[q[i]]--; dp[q[i]]+=dp[x];
                if(dp[q[i]]>=mod) dp[q[i]]-=mod;
                if(du[q[i]]==0) Q[++rg]=q[i];
            }
        }
        bool noans=0; int ret=0;
        for(i=0;i<=k;i++) if(du[id(n,i)]) noans=1;
        for(i=0;i<=k;i++) {ret+=dp[id(n,i)]; if(ret>=mod) ret-=mod;}
        if(noans) printf("-1\n");
        else printf("%d\n",ret);
    }
    return 0;
}