1. 程式人生 > 實用技巧 >CF1436E Complicated Computations(權值線段樹)

CF1436E Complicated Computations(權值線段樹)

題意:

給一個長度為n的陣列,詢問這個陣列的所有子串的MEX組成的序列的MEX

題解:

權值線段樹,列舉每個數每個出現的位置和上次出現的位置之間的區間,看看這個區間的1到a[i]-1是否全部出現,這個只要維護每個數當前最後出現的位置,然後看看是否有數的位置小於上次出現的位置就好。

這一步用權值線段樹就能完成。

//求一個序列的所有子序列的MEX組成的序列的MEX
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+100;
int a[maxn];
int pre[maxn];
int ans[maxn];
int
n; struct node { int l,r,sum; }segTree[maxn<<2]; void build (int i,int l,int r) { segTree[i].l=l; segTree[i].r=r; if (l==r) { segTree[i].sum=0; return; } int mid=(l+r)>>1; build(i<<1,l,mid); build(i<<1|1,mid+1,r); segTree[i].sum
=min(segTree[i<<1].sum,segTree[i<<1|1].sum); } void up (int i,int p,int v) { if (segTree[i].l==p&&segTree[i].r==p) { segTree[i].sum=v; return; } int mid=(segTree[i].l+segTree[i].r)>>1; if (p<=mid) up(i<<1,p,v); if (p>mid) up(i<<1
|1,p,v); segTree[i].sum=min(segTree[i<<1].sum,segTree[i<<1|1].sum); } int query (int i,int l,int r) { if (segTree[i].l>=l&&segTree[i].r<=r) { return segTree[i].sum; } int mid=(segTree[i].l+segTree[i].r)>>1; int ans=1e9; if (l<=mid) ans=min(ans,query(i<<1,l,r)); if (r>mid) ans=min(ans,query(i<<1|1,l,r)); return ans; } int main () { scanf("%d",&n); for (int i=1;i<=n;i++) scanf("%d",a+i),ans[i]=1;ans[n+1]=1;ans[n+2]=1; for (int i=1;i<=n;i++) if (a[i]!=1) ans[1]=0; build(1,1,n); for (int i=1;i<=n;i++) { if (a[i]==1) { up(1,a[i],i); pre[a[i]]=i; continue; } int p=query(1,1,a[i]-1);//找到1到a[i]-1這個區間的最小位置,沒出現就是0 //printf("%d\n",p); if (p>pre[a[i]]) ans[a[i]]=0; up(1,a[i],i); pre[a[i]]=i; } for (int i=2;i<=n+1;i++) { int p=query(1,1,i-1); if (p>pre[i]) ans[i]=0;//如果有最小的小於pre[i] } for (int i=1;i<=n+2;i++) { if (ans[i]) { printf("%d\n",i); break; } } }