我连D1T3都不会我联赛完蛋了

题目描述

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

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

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

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

为避免输出过大,答案对 P 取模。

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


输入

第一行包含一个整数 T , 代表数据组数。

接下来 T 组数据,对于每组数据: 第一行包含四个整数 N,M,K,P ,每两个整数之间用一个空格隔开。

接下来 M 行,每行三个整数 ai,bi,ci ,代表编号为 ai,bi 的点之间有一条权值为 ci 的有向边,每两个整数之间用一个空格隔开(可能有0边)。

对于 100%的数据, 1≤ P ≤ 10,N ≤ 100000 ,M ≤ 200000 ,1 ≤ ai,bi ≤ N,0 ≤ ci ≤ 1000。


输出

输出文件包含 T 行,每行一个整数代表答案。

 

$Solution:$

30分是裸的spfa最短路计数(要搁我估计拿了就歇B了)

注意到K数据范围很小,考虑把它放到状态里跑dp。

首先建反图跑一下最短路,这样我们就得到了所有点到n的最短距离。

有什么用?转移的时候就可以知道如果走这条边,会对“比最短路多走的量”作出多少贡献。

设$f[x][k]$为x到n的距离不超过x的n最短路距离+k的方案数,可得转移:

$f[x][k]=\sum f[y][k-(d(y,n)-d(x,n)+w[i])]$,$d(y,n)$表示y到n的最短路长度。

记忆化搜索即可。用一个二维数组记录一下该状态是否在栈中,用来判0环。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
#define pa pair<int,int>
using namespace std;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return x*f;
}
typedef long long ll;
const int N=1e5+5,M=4e5+5;
int K,T,n,m,dis[N],v[N],ins[N][55];
ll mod,f[N][55];
namespace G
{
    int to[M],head[N],nxt[M],w[M],tot;
    void add(int x,int y,int z)
    {
        to[++tot]=y;
        nxt[tot]=head[x];
        head[x]=tot;
        w[tot]=z;
    }
}
namespace rev
{
    int to[M],head[N],nxt[M],w[M],tot;
    void add(int x,int y,int z)
    {
        to[++tot]=y;
        nxt[tot]=head[x];
        head[x]=tot;
        w[tot]=z;
    }
}
void dj()
{
    priority_queue<pa> q;
    memset(dis,0x3f,sizeof(dis));
    memset(v,0,sizeof(v));
    dis[n]=0;
    q.push(make_pair(0,n));
    while(!q.empty())
    {
        int x=q.top().second;q.pop();
        if(v[x])continue;
        v[x]=1;
        for(int i=rev::head[x];i;i=rev::nxt[i])
        {
            int y=rev::to[i];
            if(dis[y]>dis[x]+rev::w[i])
                dis[y]=dis[x]+rev::w[i],q.push(make_pair(-dis[y],y));
        }
    }
}
ll dfs(int x,int k)
{
    if(ins[x][k])return -1;
    if(f[x][k])return f[x][k];
    ins[x][k]=1;
    if(x==n)f[x][k]=1;
    for(int i=G::head[x];i;i=G::nxt[i])
    {
        int y=G::to[i],now=dis[y]-dis[x]+G::w[i];
        if(now<=k)
        {
            ll ret=dfs(y,k-now);
            if(ret==-1)return f[x][k]=-1;
            (f[x][k]+=ret)%=mod;
        }
    }
    ins[x][k]=0;
    return f[x][k];
}
void work()
{
    n=read();m=read();K=read();mod=read();
    for(int i=1;i<=n;i++)
        G::head[i]=rev::head[i]=v[i]=0;
    G::tot=rev::tot=0;
    for(int i=1;i<=m;i++)
    {
        int x=read(),y=read(),z=read();
        G::add(x,y,z);
        rev::add(y,x,z);
    }
    dj();
    memset(f,0,sizeof(f));
    memset(ins,0,sizeof(ins));
    printf("%lld\n",dfs(1,K));
}
int main()
{
    T=read();
    while(T--)work();
    return 0;
}
01-23 19:02