【NOIP2017提高组】逛公园

                                             逛公园

题目背景

NOIP2017提高组 DAY1 T3

题目描述

策策同学特别喜欢逛公园。公园可以看成一张 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 的有向边，每两个整数之间用一个空格隔开。

输出格式

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

样例数据 1

输入　 [复制]

2 
5 7 2 10 
1 2 1 
2 4 0 
4 5 2 
2 3 2 
3 4 1 
3 5 2 
1 5 3 
2 2 0 10 
1 2 0 
2 1 0

输出

3 
-1

备注

【输入输出样例1说明】
对于第一组数据，最短路为 3 。
1-5，1-2-4-5，1-2-3-5为3条合法路径。

【数据规模与约定】
对于不同的测试点，我们约定各种参数的规模不会超过如下

对于 100% 的数据，1≤P≤10^9；1≤ai,bi≤N；0≤ci≤1000。
数据保证：至少存在一条合法的路线。

 

解析：

       首先肯定要求最短路（dijkstra/SPFA），然后就变成DP了。

       令 f[ i ][ j ] 表示到点 i 比 dis[ i ] 大 j 的的路径条数，于是就有：

       

       于是就可以进行记忆化搜索了。

       解释一下，看到 dis[ i ] − dis[ k ] + j − len[ i ][ k ] 是不是觉得有点蒙，其实是由这个等式转化而来的：

       x + dis[ k ] +len[ i ][ k ] - dis[ i ] = j，其中 x 表示如果到点 i 比 dis[ i ] 大 j ，那么到点 k 应该比 dis[ k ] 大多少 。所以其实 f[ k ][dis[ i ] − dis[ k ] + j − len[ i ][ k ]] 表示的是 f[ k ][ x ]。

       对于判断0环，那么就开个数组记录一下 f[ i ][ j ] 是否已经存在，若存在直接跳过。

 

代码：

#include <bits/stdc++.h>
using namespace std;

const int Maxn=100010;
const int Maxm=200010;
int t,n,m,size,k,mod,ans,minn,flag;
int dis[Maxn],First[Maxn],first[Maxn],f[Maxn][55],vis[Maxn][55];
struct shu{int to,next,len;};
shu Edge[Maxm],edge[Maxm];
priority_queue<pair<int,int> >q;

inline int get_int()
{
	int x=0,f=1;
	char c;
	for(c=getchar();(!isdigit(c))&&(c!='-');c=getchar());
	if(c=='-') {f=-1;c=getchar();}
	for(;isdigit(c);c=getchar()) x=(x<<3)+(x<<1)+c-'0';
	return x*f;
}

inline void clean()
{
	size=ans=flag=0;
	memset(dis,0x3f3f3f,sizeof(dis));
	memset(First,0,sizeof(First));
	memset(first,0,sizeof(first));
	memset(f,-1,sizeof(f));
	memset(vis,0,sizeof(vis));
}

inline void build(int x,int y,int z)
{
	Edge[++size].next=First[x];
	First[x]=size;
	Edge[size].to=y,Edge[size].len=z;
	edge[size].next=first[y];
	first[y]=size;
	edge[size].to=x,edge[size].len=z;
}

inline void dijkstra()
{
	q.push(make_pair(0,1)),dis[1]=0;
	while(!q.empty())
	{
	  int point=q.top().second;
	  q.pop();
	  for(int u=First[point];u;u=Edge[u].next)
	  {
	  	int to=Edge[u].to;
	  	if(dis[to] > dis[point] + Edge[u].len)
	  	{
	  	  dis[to]=dis[point] + Edge[u].len;
	  	  q.push(make_pair(-dis[to],to));
	  	}
	  }
	}
}

inline int dfs(int point,int k)
{
	if(~f[point][k]) return f[point][k];
	f[point][k]=0,vis[point][k]=1;
	for(register int u=first[point];u;u=edge[u].next)
	{
	  int to=edge[u].to,num=dis[point]-dis[to]+k-edge[u].len;
	  if(num<0) continue;
	  if(vis[to][num]) flag=1;
	  f[point][k]=(f[point][k]+dfs(to,num))%mod;
	}
	vis[point][k]=0;
	return f[point][k];
}

int main()
{
	t=get_int();
	while(t--)
	{
	  clean();
	  n=get_int(),m=get_int(),k=get_int(),mod=get_int();
	  for(int i=1;i<=m;i++)
	  {
	  	int x=get_int(),y=get_int(),z=get_int();
	  	build(x,y,z);
	  }
	  dijkstra();
	  if(!dis[n]) {cout<<"-1\n";continue;}
	  f[1][0]=1;
	  for(int i=0;i<=k;i++) ans=(ans+dfs(n,i))%mod;
	  if(!flag) cout<<ans<<"\n";
	  else cout<<"-1\n";
	}
	return 0;
}