2018.11.01 洛谷P3953 逛公园(最短路+dp)

传送门
f [ i ] [ j ] f[i][j] f[i][j]表示跟最短路差值为 i i i当前在点 j j j的方案数。
i n [ i ] [ j ] in[i][j] in[i][j]表示在被选择的集合当中。
大力记忆化搜索就行了。
代码:

#include
using namespace std;
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
const int N=1e5+5,M=2e5+5;
int d[N],T,n,m,k,mod,first[N],cnt=0,ans,flag,f[55][N],dis[N];
bool in[55][N],vis[N];
struct edge{int v,w,next;}e[M<<1];
inline void add(int u,int v,int w){e[++cnt].v=v,e[cnt].w=w,e[cnt].next=first[u],first[u]=cnt;}
inline void spfa(){
	queue<int>q;
	dis[1]=0,vis[1]=1,q.push(1);
	while(!q.empty()){
		int x=q.front();
		q.pop(),vis[x]=0;
		for(int i=first[x];i;i=e[i].next){
			int v=e[i].v;
			if(dis[v]>dis[x]+e[i].w){
				dis[v]=dis[x]+e[i].w;
				if(!vis[v])vis[v]=1,q.push(v);
			}
		}
	}
}
inline int dfs(int p,int d){
	if(flag)return 0;
	if(in[d][p])return flag=1,0;
	if(~f[d][p])return f[d][p];
	in[d][p]=1,f[d][p]=(p==n&&d<=k);
	for(int i=first[p];i;i=e[i].next){
		int v=e[i].v,dist=dis[p]+e[i].w+d-dis[v];
		if(dist<0||dist>k)continue;
		f[d][p]+=dfs(v,dist);
		if(f[d][p]>=mod)f[d][p]-=mod;
	}
	return in[d][p]=0,f[d][p];
}
int main(){
	T=read();
	while(T--){
		n=read(),m=read(),k=read(),mod=read();
		memset(first,0,sizeof(first)),memset(dis,127,sizeof(dis)),memset(f,-1,sizeof(f)),memset(in,0,sizeof(in));
		flag=ans=cnt=0;
		for(int i=1,u,v,w;i<=m;++i)u=read(),v=read(),w=read(),add(u,v,w);
		spfa(),ans=dfs(1,0),printf("%d\n",flag?-1:ans);
	}
	return 0;
}

你可能感兴趣的:(#,最短路,#,dp)