GYM 100090 M.Jumping along the Hummocks（SPFA）

Description
n个点，每个点有颜色，i点到i+1点距离为1，一个点到后面距其最近且颜色与其相同的点的距离也为1，求1点到n点的最短距离
Input
第一行一整数n表示点数，之后n个整数c[i]表示第i个点的颜色(1<=n<=200000,1<=c[i]<=200000)
Output
输出一整数表示1到n的最短距离
Sample Input
6
1 8 1 1 7 1
Sample Output
3
Solution
建完图跑一边最短路即可
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 222222
#define maxm 444444
struct edge
{
    int to,next;
    int cost;
}g[maxm];
int head[maxn],tol;
int dis[maxn];//所有点到起点的最短距离
void init()//初始化 
{
    memset(head,-1,sizeof(head));
    tol=0;
}
void add(int u,int v,int c)//单向边,从u到v,权值为c 
{
    g[tol].cost=c;
    g[tol].to=v;
    g[tol].next=head[u];
    head[u]=tol++;
}
void spfa(int s)//单源最短路,s是起点
{
    bool vis[maxn];
    memset(vis,false,sizeof(vis));
    queue<int>que;
    for(int i=0;i<maxn;i++)
        dis[i]=INF;
    dis[s]=0;
    vis[s]=true;
    que.push(s);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        vis[u]=false;
        for(int i=head[u];i!=-1;i=g[i].next)
        {
            int v=g[i].to;
            int c=g[i].cost;
            if(dis[v]>dis[u]+c)
            {
                dis[v]=dis[u]+c;
                if(!vis[v])
                {
                    vis[v]=true;
                    que.push(v);
                }
            }
        }
    }
}
int n,c[maxn];
vector<int>v[maxn];
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=200000;i++)v[i].clear();
        for(int i=1;i<=n;i++)
            scanf("%d",&c[i]),v[c[i]].push_back(i);
        init();
        for(int i=1;i<=n;i++)add(i,i+1,1);
        for(int i=1;i<=200000;i++)
        {
            if(v[i].size()<2)continue;
            int pre=v[i][0];
            for(int j=1;j<v[i].size();j++)
            {
                int next=v[i][j];
                add(pre,next,1);
                pre=next;
            }
        }
        spfa(1);
        printf("%d\n",dis[n]);
    }
    return 0;
}