我的最短路模板
dijkstra(链式前向星+优先队列优化)
#include<stdio.h>
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=1e7;
//链式前向星
int cnt;
int dis[1000];
int path[1000];
struct Dis
{
int n;
int len;
Dis(int x):n(x),len(dis[x]){};
};
bool operator<(const Dis& a,const Dis& b)
{
return a.len<b.len;
}
struct Edge
{
int next;//与第i条边相同起点相同的下一条边
int to;// 第i条变的终点
int w;//第i条边的权值
}edge[1000];
int head[1000];//起点为i的第一条边(与输入逆序)
void add(int u,int v,int w)//加入新的截点
{
edge[cnt].to=v;
edge[cnt].next=head[u];
edge[cnt].w=w;
head[u]=cnt++;
// printf("to=%d next=%d head[%d]=%d\n",edge[cnt-1].to,edge[cnt-1].next,u,head[u]);
}
void dijkstra(int u,int v,int n)
{
//u是起点,v是终点,n为图里面点的个数
priority_queue<Dis> Q;
fill(dis,dis+n+1,maxn);
memset(vis,0,sizeof vis);
dis[u]=0;
Q.push(Dis(u));
while(!Q.empty())
{
Dis temp=Q.top();
Q.pop();
if(vis[temp.n])continue;
vis[temp.n]=1;
// cout<<temp.n<<" "<<temp.len<<endl;
for(int i=head[temp.n];i!=-1;i=edge[i].next)
{
// cout<<"i="<<i<<endl;
if(dis[temp.n]+edge[i].w<dis[edge[i].to])
{
// cout<<"-------"<<endl;
dis[edge[i].to]=dis[temp.n]+edge[i].w;
path[edge[i].to]=temp.n;
Q.push(Dis(edge[i].to));
}
}
}
}
int main()
{
int n,a,b;
while(cin>>n&&n)
{
cin>>a>>b;
cnt=0;
fill(head,head+1000,-1);
for(int i=1;i<=n;i++)
{
int len;
cin>>len;
if(i+len<=n)
{
add(i,i+len,1);
}
if(i-len>=1)
{
add(i,i-len,1);
}
}
dijkstra(a,b,n);
int ans=(dis[b]==maxn?-1:dis[b]);
cout<<ans<<endl;
}
}
无链式前向星版
vector<pii> g[maxn];
bool vis[maxn];
struct Dis{
int n,len;
Dis(int x,int L):n(x),len(L){};
bool operator<(const Dis& b)const{
return len>b.len;
}
};
void dijkstra(int s,int dis[]){
priority_queue<Dis> Q;
fill(dis,dis+n+1,maxn);
memset(vis,0,sizeof vis);
dis[s]=0;
Q.push(Dis(s,0));
while(!Q.empty()){
Dis now=Q.top();
Q.pop();
int u=now.n;
if(vis[u])continue;
vis[u]=1;
for(auto v:g[u]){
if(dis[u]+v.se<dis[v.fi]){
dis[v]=dis[u]+v.se;
Q.push(Dis(v,dis[v.fi]));
}
}
}
}
Q.E.D.