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 号点转移的状态就全部是合法的,依此类推。

NOIP 2017 逛公园 记忆化搜索 最短路 好题

考虑正着进行记忆化搜索,从一个点转移到另一个点时,所需要走的最短路径为 $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;
}

  

上一篇:Codeforces 258D Little Elephant and Broken Sorting (看题解) 概率dp


下一篇:HDU - 5001 Walk(概率dp+记忆化搜索)