dijisktra
将所有在第i层可能的移动路径存储到邻接表中,然后跑一遍迪杰斯特拉,细节见注释
C++ 代码
#include<bits/stdc++.h>
using namespace std;
const int N=2e4+10,M=1001;
int n,m,s;
int c[21],ans;
int fist[N],nxt[N],go[N],w[N],tot;//w存储把手位
void add(int x,int y,int z)
{
nxt[++tot]=fist[x];
fist[x]=tot;
go[tot]=y;
w[tot]=z;
}
priority_queue<pair<int,int> >q;
int d[M],cz[M];//cz->当前把手位
bool v[M];
void diji()
{
memset(d,0x3f,sizeof d);
d[1]=0;cz[1]=s;
q.push(make_pair(0,1));
while(q.size())
{
int x=q.top().second;q.pop();
if(v[x]) continue;v[x]=1;
int now=cz[x];
for(int i=fist[x];i;i=nxt[i])
{
int y=go[i];
if(d[y]>d[x]+2*abs(y-x)+abs(w[i]-now))//移动一层需要2个时间,移动一个把手位需要1个时间
{
d[y]=d[x]+2*abs(y-x)+abs(w[i]-now);
cz[y]=w[i];
q.push(make_pair(-d[y],y));
}
}
}
if(d[n]==0x3f3f3f3f) puts("-1");
else cout<<d[n];
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
scanf("%d",&c[i]);
if(!c[i]) s=i;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
if(i+c[j]<=0||i+c[j]>n||!c[j]) continue;
add(i,i+c[j],j);//i->当前位置,
//i+c[j]->移动后的位置,
//j->此时把手该处于哪个位置
}
diji();
return 0;
}