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