NOIP 2017 逛公园 记忆化搜索 最短路 好题
题目描述:
策策同学特别喜欢逛公园。公园可以看成一张N个点MM条边构成的有向图,且没有 自环和重边。其中1号点是公园的入口,N号点是公园的出口,每条边有一个非负权值, 代表策策经过这条边所要花的时间。
策策每天都会去逛公园,他总是从1号点进去,从N号点出来。
策策喜欢新鲜的事物,它不希望有两天逛公园的路线完全一样,同时策策还是一个 特别热爱学习的好孩子,它不希望每天在逛公园这件事上花费太多的时间。如果1号点 到N号点的最短路长为d,那么策策只会喜欢长度不超过d + K的路线。
策策同学想知道总共有多少条满足条件的路线,你能帮帮它吗?
为避免输出过大,答案对P取模。
如果有无穷多条合法的路线,请输出-1。
题解:
话说去年提高组的难度真心大.......
我们一层一层考虑这道题:
1.首先,我们发现 $k<=50$,这个规模是非常小的。
2.由发现一,我们不难列出 dp 状态:$dp[i][j]$ 代表第 $i$ 号点到终点的距离比最短路大 $j$ 的方案数。
对于2的实现,我们考虑将每条边反着连
从N 点开始跑一遍到 1 号点的最短路,求出每个点到 N 号点的最短路。
我们这么做的原因是由于正着做可能会碰到不合法的状态,而逆推则不会碰到非法状态。
因为显然,1 号点的最短路径一定是全局的最短路径,于是可以由 1 号点转移的状态就全部是合法的,依此类推。
考虑正着进行记忆化搜索,从一个点转移到另一个点时,所需要走的最短路径为 $d[v]+w$。
而原本的最短路径为 $d[u]$,那么这就比 $d[u]$ 多了 $d[v]+w-d[u]$ 的路程。
假设当前的偏移量(即比最短路多出的量)为 $k$。
那么转移到 $v$ 点之后的偏移量就会变成 $k-(d[v]+w-d[u])$
我们这么进行记忆化搜索即可。
不过我们还要判一下 0 环。 开一个 $vis[i][k]$ 数组即可。
发现该状态再一次被访问且 $vis[i][k]=true$ 就说明出现 0 环,直接输出 -1.
Code:
#include<cstdio> #include<cstring> #include<algorithm> #include<queue> #include<string> using namespace std; const int maxn=400000+4; int d[maxn],s, n ,m, mod,tot; struct SPFA{ int head[maxn],to[maxn],nex[maxn],val[maxn],cnt; void addedge(int u,int v,int c){ nex[++cnt]=head[u],head[u]=cnt,to[cnt]=v,val[cnt]=c; } queue<int>Q; bool inq[maxn]; void init(){ memset(head,0,sizeof(head)), memset(to,0,sizeof(to)), memset(nex,0,sizeof(nex)), memset(val,0,sizeof(val)); cnt=0; } void spfa(){ memset(d,0x3f,sizeof(d)); memset(inq,false,sizeof(inq)); d[s]=0,inq[s]=true; Q.push(s); while(!Q.empty()){ int u=Q.front();Q.pop(); inq[u]=false; for(int v=head[u];v;v=nex[v]){ if(d[u]+val[v]<d[to[v]]){ d[to[v]]=d[u]+val[v]; if(!inq[to[v]]){ Q.push(to[v]); inq[to[v]]=true; } } } } } }T; int head[maxn], to[maxn], nex[maxn], val[maxn], cnt; void addedge(int u,int v,int c){ nex[++cnt]=head[u], head[u]=cnt, to[cnt]=v, val[cnt]=c; } int dp[100007][60]; bool vis[100007][60]; int dfs(int u,int k){ if(vis[u][k]) return -1; if(dp[u][k]!=-1) return dp[u][k]; vis[u][k]=1; int sum=0; for(int v=head[u];v;v=nex[v]){ int tmp=k-(d[to[v]]+val[v]-d[u]); if(tmp<0||tmp>tot) continue; int delta=dfs(to[v], tmp); if(delta==-1) return -1; sum=(sum+delta)%mod; } if(k==0&&u==n) sum+=1; vis[u][k]=0; dp[u][k]=sum; return sum; } int work(){ memset(dp,-1,sizeof(dp)); memset(head,0,sizeof(head)); memset(to,0,sizeof(to)); memset(val,0,sizeof(val)); cnt=0; T.init(); scanf("%d%d%d%d",&n,&m,&tot,&mod); for(int i=1;i<=m;++i){ int a,b,c; scanf("%d%d%d",&a,&b,&c); T.addedge(b,a,c); // 返向 addedge(a,b,c); // 正向 } s=n; T.spfa(); int ans=0; for(int i=0;i<=tot;++i){ memset(vis,0,sizeof(vis)); int aa=dfs(1,i); if(aa==-1) return -1; ans=(ans+aa)%mod; } return ans; } int main(){ int T; scanf("%d",&T); while(T--) printf("%d\n",work()); return 0; }