CodeForces 689B Mike and Shortcuts spfa最短路

CF我做的不是很多，但最好的一點是官網上可以看到評測點，方便找問題。這道題一看就是最短路，因爲n有些大，dijkstra可能會T，就用了spfa(不用判負環)。一開始寫的不是TLE就是MLE，怎麼都想不通哪裏有問題，後來看了大佬博客Orzhttp://www.21ic.com/tougao/article/13224.html，猛然發現邊加多了。如果不算捷徑，除1以外的其他節點直接到節點1的距離和通過他的前繼到達節點1的距離是一樣的，利用這個就可以省去大部分邊，只保留1->2->3->....->n的這種邊和捷徑，長度都是1，降低了大量的空間和時間複雜度。附上AC代碼：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#define pb push_back
using namespace std;
const int INF=0x3f3f3f3f;

const int MAXN=200002;
int n;
vector<int>edge[MAXN];//只存儲點,邊長都是1
int dis[MAXN];
bool vis[MAXN];
void spfa(int sta)
{
    memset(vis,false,sizeof(vis));
    for(int i=0;i<=n;i++) dis[i]=INF;
    dis[sta]=0;vis[sta]=true;
    queue<int>que;
    while(!que.empty()) que.pop();
    que.push(sta);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        vis[u]=false;//注意!!!
        for(int i=0;i<edge[u].size();i++)
        {
            int v=edge[u][i];
            int w=1;
            if(dis[v]>dis[u]+w)
            {
                dis[v]=dis[u]+w;
                if(!vis[v])
                {
                    vis[v]=true;
                    que.push(v);
                }
            }
        }
    }
}

int main()
{
    while(scanf("%d",&n)==1)
    {
        for(int i=0;i<=n;i++)
            edge[i].clear();
        int x;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            if(x!=i)
                edge[i].pb(x);
        }
        for(int i=1;i<n;i++)
        {
            edge[i].pb(i+1);
            edge[i+1].pb(i);//這種邊雙向
        }
        spfa(1);
        for(int i=1;i<n;i++)
            printf("%d ",dis[i]);
        printf("%d\n",dis[n]);
    }
	return 0;
}