正題

題目鏈接:https://www.luogu.com.cn/problem/AT4439

---

題目大意

給出$1～n$的排列$a$。求一個(gè)字典序最小的$01$串$s$滿(mǎn)足將$0$對(duì)應(yīng)位置按順序取出成為序列$A$，剩下的成為序列$B$。

要求$A$和$B$的前綴最大值個(gè)數(shù)相同。

$1\le n\le 2\times 10^5$

---

解題思路

首先對(duì)于前綴最大值來(lái)說(shuō)，在排列$a$中的前綴最大值肯定在$A/B$中也是前綴最大值。

而假設(shè)我們序列$A$和$B$中都存在一個(gè)前綴最大值是在$a$中沒(méi)有出現(xiàn)過(guò)的，那么顯然這兩個(gè)值前面比它大的值都在另一個(gè)序列中，所以我們交換這兩個(gè)值時(shí)$A/B$的前綴最大值個(gè)數(shù)都減少了$1$。

所以如果存在一組解$A/B$中存在一個(gè)序列的所有前綴最大值都是$a$中原來(lái)的最大值。

那么接著考慮，假設(shè)我們做到一個(gè)狀態(tài)：$A/B$中目前最大值個(gè)數(shù)為$n_a/n_b$，后面還有$c$個(gè)原來(lái)$a$序列中的最大值，$B$需要用$k$個(gè)，剩下$p$個(gè)都是新的最大值，那么如果有解就有等式
$$n_a+c?k=n_b+k+p?n_a?n_b+c=2k+p$$
而左邊的式子是定值，所以我們只需要考慮右邊式子的取值范圍。

然后對(duì)于序列$B$，目前最后一個(gè)數(shù)是$m_b$，設(shè)舊最大值數(shù)權(quán)為$2$，其他數(shù)的權(quán)值$1$。我們就需要考慮后面是否存在一個(gè)$m_b$開(kāi)始的上升序列的權(quán)值和為$n_a?n_b+c$。

而因?yàn)闄?quán)值只有$1$和$2$，所以我們用數(shù)據(jù)結(jié)構(gòu)維護(hù)一下奇偶的最大答案即可。

由于他要求字典序最小，我們無(wú)法確定$A$是全是舊的最大值還是$B$全是舊的最大值，所以我們兩種情況都需要判斷。

時(shí)間復(fù)雜度：$O(n\log n)$

---

code

#include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int N=2e5+10; int n,s,a[N],od[N],ans[N],f[N][2]; struct SegTree{int w[N<<2];void Change(int x,int L,int R,int pos,int val){if(L==R){w[x]=val;return;}int mid=(L+R)>>1;if(pos<=mid)Change(x*2,L,mid,pos,val);else Change(x*2+1,mid+1,R,pos,val);w[x]=max(w[x*2],w[x*2+1]);}int Ask(int x,int L,int R,int l,int r){l=max(l,L);r=min(r,R);if(l>r)return w[0];if(L==l&&R==r)return w[x];int mid=(L+R)>>1;if(r<=mid)return Ask(x*2,L,mid,l,r);if(l>mid)return Ask(x*2+1,mid+1,R,l,r);return max(Ask(x*2,L,mid,l,mid),Ask(x*2+1,mid+1,R,mid+1,r));} }T[2]; bool check(int p,int x){if(x<0)return 0;return T[x&1].Ask(1,1,n,p,n)>=x; } int main() {scanf("%d",&n);int maxs=0;for(int i=1;i<=n;i++){scanf("%d",&a[i]);if(a[i]>maxs)od[i]=1,s++,maxs=a[i]; }memset(T[1].w,0xcf,sizeof(T[1].w));for(int i=n;i>=1;i--){int p=!od[i];for(int j=0;j<2;j++){f[i][j]=T[j^p].Ask(1,1,n,a[i]+1,n)+1+od[i];T[j].Change(1,1,n,a[i],f[i][j]);}}int A=0,B=0,ma=0,mb=0;for(int i=1;i<=n;i++){s-=od[i];T[0].Change(1,1,n,a[i],T[0].w[0]);T[1].Change(1,1,n,a[i],T[1].w[0]);if(check(max(ma,a[i]),B+s-A-(a[i]>ma))||check(mb,A+s-B+(a[i]>ma)))ans[i]=0,A+=(a[i]>ma),ma=max(ma,a[i]);else ans[i]=1,B+=(a[i]>mb),mb=max(mb,a[i]);}if(A!=B)return puts("-1")&0;for(int i=1;i<=n;i++)putchar(ans[i]+'0');return 0; }

總結(jié)

以上是生活随笔為你收集整理的AT4439-[AGC028E]High Elements【结论,线段树】的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。