[FROM LUOGU]逛公园

传送门

SOL
这道题的记忆化搜索、最短路DP有很多被卡了的,原因在于没有考虑0边不在路径上的情况,或者按照0边建图跑拓扑导致非0环上的点被误伤
这个分层图(其实还是DP)的做法没有被卡:
跑拓扑我们在最短路边集(所有dis<=K的路径上的边)上进行(当然你得先跑正反图的最短路)
那么显然非0环至少有一条边不会被加进去,
而0环会,这也就是导致 − 1 -1 1的原因,所以跑完拓扑判断一下有没有入度还不为0的点即可
接下来跑分层图(最短路DP)即可
我们先把边权处理成相对于最短路的增量,然后按照层数由小到大更新即可

代码:

#include<bits/stdc++.h>
using namespace std;
#define re register
#define ll long long
inline int rd(){
	int re data=0,w=1;static char ch=0;ch=getchar();
	while(!isdigit(ch)&&ch!='-')ch=getchar();
	if(ch=='-')w=-1,ch=getchar();
	while(isdigit(ch))data=(data<<1)+(data<<3)+(ch^48),ch=getchar();
	return data*w;
}
const int N=1e5+5,M=2e5+5,inf=1e9;
struct edges{int v,w,nxt;}e[M];
typedef pair<int,int> T;
#define mp make_pair
priority_queue<T,vector<T>,greater<T> >q;
int Test_case,n,m,k,p,first[N],ts[N],stk[N],cnt,dp[N][51],du[N],a[M],b[M],c[M],f[N],g[N];bool vis[N];
inline void add(int u,int v,int w){e[++cnt]=(edges){v,w,first[u]},first[u]=cnt;}
inline void dijkstra(int st,int *dis){
	for(int re i=1;i<=n;++i)dis[i]=inf;dis[st]=0,q.push(mp(0,st));
	while(!q.empty()){
		T t=q.top();q.pop();
		int re d=t.first,u=t.second;
		if(dis[u]!=d)continue;
		for(int re i=first[u];i;i=e[i].nxt){
			int re v=e[i].v,w=e[i].w;
			if(dis[u]+w<dis[v])dis[v]=dis[u]+w,q.push(mp(dis[v],v));
		}
	}
}
inline void topsort(){
	int re top=1,pos=0;stk[top]=1;
	while(top>0){
		int re u=stk[top--];
		ts[++pos]=u;
		for(int re i=first[u];i;i=e[i].nxt){
			int re v=e[i].v;
			--du[v];if(!du[v])stk[++top]=v;
		}
	}
}
inline void init(){cnt=0,memset(first,0,sizeof first),memset(vis,0,sizeof vis),memset(ts,0,sizeof(ts)),memset(du,0,sizeof(du)),memset(dp,0,sizeof(dp));}
signed main(){
	Test_case=rd();
	while(Test_case--){
		n=rd(),m=rd(),k=rd(),p=rd(),init();
		for(int re i=1;i<=m;++i)a[i]=rd(),b[i]=rd(),c[i]=rd(),add(a[i],b[i],c[i]);
		dijkstra(1,f),cnt=0,memset(first,0,sizeof(first));
		for(int re i=1;i<=m;++i)add(b[i],a[i],c[i]);
		dijkstra(n,g),cnt=0,memset(first,0,sizeof(first));
		for(int re i=1;i<=m;++i)
			if(f[a[i]]+g[b[i]]+c[i]<=f[n]+k&&f[a[i]]+c[i]==f[b[i]])
				add(a[i],b[i],0),++du[b[i]];
		for(int re i=1;i<=m;++i)c[i]=f[a[i]]+c[i]-f[b[i]];
		topsort();
		bool re fl=1;
		for(int re i=1;i<=n;++i)if(du[i])fl=0;
		if(!fl){puts("-1");continue;}
		dp[1][0]=1;
		for(int re j=0;j<=k;++j){
			for(int re t=1;t<=n;++t){
				int re u=ts[t];
				for(int re v,i=first[u];i;i=e[i].nxt)
					v=e[i].v,(dp[v][j]+=dp[u][j])%=p;
			}			
			for(int re i=1;i<=m;++i)if(j+c[i]<=k&&c[i])(dp[b[i]][j+c[i]]+=dp[a[i]][j])%=p;
		}
		int re ans=0;
		for(int re i=0;i<=k;++i)(ans+=dp[n][i])%=p;
		printf("%d\n",ans);
	}return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值