NOIP2017D1T3逛公园 拆点+最短路+DP

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

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

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

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

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

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

题解：

这道题还是挺不错的，对于我这种蒟蒻来说，对大神来说这道题就是套路题了。把每个点拆成k+1个点，分别为0~k，第i个点表示到这个点的时间比最短路延迟了i，然后若图中没有0环，那么这样重新建出来的图就一定是个拓扑图，直接DP就好了。对于判断0环的方法，有很多，但是，我代码中的方法是错的，对于拓扑排序后的点，若还有入度，只能证明它是一个0环，但是合法的路径不一定经过他，所以还要判断环上的点在不在最短路上，这个就随便搞了，比如再从终点跑一次最短路就是一种方法。

代码：

#include<bits/stdc++.h>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;
#define LL long long
#define pa pair<int,int>
using namespace __gnu_pbds;
typedef __gnu_pbds::priority_queue<pa,greater<pa>,pairing_heap_tag > heap;
const int Maxn=100002,Maxm=200002,Maxk=53;
const int inf=2147483647;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int n,m,k,p;
struct Edge{int y,d,next;}e[Maxm],E[Maxm*Maxk];
int len,last[Maxn],Len,Last[Maxn*Maxk];
void ins(int x,int y,int d){int t=++len;e[t].y=y;e[t].d=d;e[t].next=last[x];last[x]=t;}
void Ins(int x,int y){int t=++Len;E[t].y=y;E[t].next=Last[x];Last[x]=t;}
int f[Maxn];bool vis[Maxn];
void dijkstra()
{
    heap q;q.push(make_pair(0,1));
    memset(vis,false,sizeof(vis));
    memset(f,63,sizeof(f));f[1]=0;
    while(!q.empty())
    {
        pa t=q.top();q.pop();
        int x=t.second;
        if(vis[x])continue;vis[x]=true;
        for(int i=last[x];i;i=e[i].next)
        {
            int y=e[i].y;
            if(f[x]+e[i].d<f[y])
            {
                f[y]=f[x]+e[i].d;
                q.push(make_pair(f[y],y));
            }
        }
    }
}
int deg[Maxn*Maxk],ans[Maxn*Maxk];
int main()
{
    int T=read();
    while(T--)
    {
        memset(last,0,sizeof(last));len=0;
        n=read(),m=read(),k=read(),p=read();
        for(int i=1;i<=m;i++)
        {
            int x=read(),y=read(),d=read();
            ins(x,y,d);
        }
        dijkstra();
        memset(Last,0,sizeof(Last));Len=0;
        memset(deg,0,sizeof(deg));
        memset(ans,0,sizeof(ans));
        for(int x=1;x<=n;x++)
        {
            for(int i=last[x];i;i=e[i].next)
            {
                int y=e[i].y,t=f[x]+e[i].d-f[y];
                for(int j=0;j<=k-t;j++)
                Ins(x+j*n,y+(j+t)*n),deg[y+(j+t)*n]++;
            }
        }
        queue<int>q;ans[1]=1%p;
        for(int i=1;i<=n*(k+1);i++)if(!deg[i])q.push(i);
        while(!q.empty())
        {
            int x=q.front();q.pop();
            for(int i=Last[x];i;i=E[i].next)
            {
                int y=E[i].y;
                ans[y]+=ans[x];
                if(ans[y]>=p)ans[y]-=p;
                if(!(--deg[y]))q.push(y);
            }
        }
        bool no=false;
        for(int i=1;i<=n*(k+1);i++)
        if(deg[i]){no=true;break;}
        if(no){puts("-1");continue;}
        int Ans=0;
        for(int i=1;i<=k+1;i++)
        {
            Ans+=ans[n*i];
            if(Ans>=p)Ans-=p;
        }
        printf("%d\n",Ans);
    }
}