【CF1023D】Array Restoration（构造，线段树）

题意：有一个长为n的序列，对其进行q次操作，第i次操作可以把连续的一段覆盖为i

现在给出操作后的序列，第i个数字为a[i]，其中有一些为0的位置可以为任意值，要求构造任意一组合法的操作后的序列

无解输出NO

n,q<=2e5，0<=a[i]<=q

思路：看不懂别人写的题解，照自己的思路写一个……

首先将a[i]从大到小排序，若a[i]已经确定则将a[i]填到i左右两端连续的0中

然后判断q有没有在填完之后的序列中出现，若没有出现则找一段连续的0覆盖成q，找不到0则无解

前面两步能将数列填满，预处理出填完之后数列中每个数值出现的第一次和最后一次出现的位置，如果中间有比他小的数字则不合法

 #include<cstdio>
 #include<cstring>
 #include<string>
 #include<cmath>
 #include<iostream>
 #include<algorithm>
 #include<map>
 #include<set>
 #include<queue>
 #include<vector>
 #include<bitset>
 using namespace std;
 typedef long long ll;
 typedef unsigned int uint;
 typedef unsigned long long ull;
 typedef pair<int,int> PII;
 typedef vector<int> VI;
 #define fi first
 #define se second
 #define MP make_pair
 #define N      210000
 #define M      51
 #define MOD 1000000007
 #define eps 1e-8
 #define pi     acos(-1)
 #define oo     1e9

 struct node
 {
     int x,y;
 }b[N];

 int t[N<<],a[N],c[N],l[N],r[N];

 bool cmp(node a,node b)
 {
     return a.x>b.x;
 }

 void pushup(int p)
 {
     t[p]=min(t[p<<],t[p<<|]);
 }

 void build(int l,int r,int p)
 {
     if(l==r)
     {
         t[p]=c[l];
         return;
     }
     int mid=(l+r)>>;
     build(l,mid,p<<);
     build(mid+,r,p<<|);
     pushup(p);
 }

 int query(int l,int r,int x,int y,int p)
 {
     if(x<=l&&r<=y) return t[p];
     int mid=(l+r)>>;
     int tmp=oo;
     if(x<=mid) tmp=min(tmp,query(l,mid,x,y,p<<));
     if(y>mid) tmp=min(tmp,query(mid+,r,x,y,p<<|));
     return tmp;
 }

 int main()
 {
     int n,q;
     scanf("%d%d",&n,&q);
     for(int i=;i<=n;i++)
     {
         scanf("%d",&a[i]);
         b[i].x=a[i];
         b[i].y=i;
     }
     for(int i=;i<=n;i++) c[i]=a[i];
     sort(b+,b+n+,cmp);
     for(int i=;i<=n;i++)
      if(b[i].x)
      {
          int j=b[i].y-;
         while(j>&&c[j]==) c[j--]=b[i].x;
         j=b[i].y+;
         while(j<=n&&c[j]==) c[j++]=b[i].x;
      }
     int flag=;
     for(int i=;i<=n;i++)
      if(c[i]==q){flag=; break;}
     if(!flag)
     {
         int k=;
         for(int i=;i<=n;i++)
          if(!a[i]){k=i; break;}
         if(!k)
         {
             printf("NO\n");
             return ;
         }
          else
          {
              while(k<=n&&a[k]==) c[k++]=q;
          }
     }

     for(int i=;i<=n;i++)
     {
         if(!l[c[i]]) l[c[i]]=i;
         r[c[i]]=i;
     }
     flag=;
     build(,n,);
     for(int i=;i<=q;i++)
     {
         int t=oo;
         if(l[i]<=r[i]&&l[i]&&r[i]) t=query(,n,l[i],r[i],);
         if(t<i){flag=; break;}
     }
     if(flag)
     {
         printf("YES\n");
         for(int i=;i<=n;i++) printf("%d ",c[i]);
     }
      else printf("NO\n");
     return ;
 }