最短路模板

我的最短路模板

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);
    dis[u]=0;
    Q.push(Dis(u));
    while(!Q.empty())
    {
        Dis temp=Q.top();
        Q.pop();
//      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;
    }
}