Gym - 101911E - Painting the Fence【set..】

参考大佬博客讲得很清楚https://blog.csdn.net/mmk27_word/article/details/85063931?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522160154998019725222461674%2522%252C%2522scm%2522%253A%252220140713.130102334…%2522%257D&request_id=160154998019725222461674&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2_allfirst_rank_ecpm_v3~pc_rank_v2-1-85063931.first_rank_ecpm_v3_pc_rank_v2&utm_term=Gym±+101911E&spm=1018.2118.3001.4187

题目

传送门

Input
4
1 2 1 2
2
2 1
Output
1 2 2 2
Input
8
7 1 7 1 23 9 23 1
4
23 4 7 1
Output
7 7 7 1 1 1 1 1

题意:先给出一个数n，下面接着是n个数代表每个位置的颜色，再给出一个数m，下面是m个颜色，对于每个颜色每次把这个颜色最左边和最右边的位置之间的颜色换成这个颜色，问：最后每个位置上的颜色分别是什么

思路：用set先把每种颜色的位置存储，每次对颜色进行操作时把它的最左边和最右边位置之间的颜色的位置除去，最后进行遍历

AC code

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
#include<sstream>
#include<queue>
#include<stack>
#include<set>
using namespace std;
#define ll long long
set<int>s[300010];
int n,m,vis[300010],a[300010];
int main()
{
    ios::sync_with_stdio(0);
    int p;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        s[a[i]].insert(i);//插入各颜色的位置
    }
    cin>>m;
    for(int i=0;i<m;i++)
    {
        int x;
        cin>>x;
        if(vis[x])continue;//操作过的颜色不用管
        if(s[x].size()<2)//位置小于2的元素不用管
        {
            vis[x]=1;
            continue;
        }
        int l=*(s[x].begin()),r=*(s[x].rbegin());
        for(int j=l+1;j<=r-1;j++)
        {
            s[a[j]].erase(j);//消去中间颜色的位置
            while(vis[a[j]]&&s[a[j]].size())
            {
                j=*(s[a[j]].begin());
                s[a[j]].erase(j);
            }

      }
 vis[x]=1;
    }
    for(int i=1;i<=300000;i++)
        if(s[i].size()>=2&&vis[i])//遍历
    {
        int l=*(s[i].begin()),r=*(s[i].rbegin());
        for(int j=l;j<=r;j++)a[j]=i;
    }
    for(int i=1;i<=n;i++) printf("%d%c",a[i]," \n"[i==n]);
}