最短路模板

我的最短路模板

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]));
            }
        }
    }
}