2018.11.01 낙곡P3953 공원구경(최단로+dp)

22688 단어 #최단로dp
전송문 설정 f[i] [j] f[i] [j] f[i] [j]는 최단거리 차치와 ii i가 현재 jj에 있는 방안 수를 나타낸다.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
"
,flag?-1:ans); } return 0; }

좋은 웹페이지 즐겨찾기