CF486E LIS of Sequence
题目大意
给你一个长度为 \(n\) 的序列 \(a_1,a_2,\cdots,a_n\),你需要把这 \(n\) 个元素分成三类:\(1,2,3\):
-
所有的最长上升子序列都不包含这个元素。
-
有但非所有的最长上升子序列包含这个元素。
-
所有的最长上升子序列都包含这个元素。
题目分析
令 \(f_i\) 表示 \(\{a_1,a_2,\cdots,a_i\}\) 的 \(\rm LIS\) 长度,\(g_i\) 表示 \(\{a_i,a_{i+1},\cdots,a_n\}\) 的 \(\rm LIS\) 长度,\(len\) 表示 \(\{a_1,a_2,\cdots,a_n\}\) 的 \(\rm LIS\) 长度。
对于 \(a_i\),若 \(f_i+g_i\neq len+1\),说明 \(a_i\) 这个点不在整个序列所有的 \(\rm LIS\) 方案里。那么 \(ans_i\gets 1\)。
若 \(f_i+g_i=len+1\),判断是否存在 \(f_i=f_j\) 且 \(g_i=g_j(i\neq j)\),则 \(a_i\) 一定属于第二种情况:有但非所有的最长上升子序列包含这个元素。因为其他元素和 \(a_i\) 某种意义上来说等价,都在其 \(\rm LIS\) 中起到同样的作用。
其他情况为最后一种情况。
\(f_i,g_i\) 可以用二分、线段树和树状数组在 \(\mathcal{O(n\log n)}\) 的时间内求得,记录是否出现重复可以用桶或 \(map\)。树状数组码量小,完爆线段树;树状数组容易理解,完爆二分。故此处选择树状数组。
代码
//2022/2/28
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstdio>
#include <climits>//need "INT_MAX","INT_MIN"
#include <cstring>//need "memset"
#include <numeric>
#include <algorithm>
#include <map>
#define enter() putchar(10)
#define debug(c,que) cerr << #c << " = " << c << que
#define cek(c) puts(c)
#define blow(arr,st,ed,w) for(register int i = (st);i <= (ed); ++ i) cout << arr[i] << w;
#define speed_up() cin.tie(0),cout.tie(0)
#define mst(a,k) memset(a,k,sizeof(a))
#define Abs(x) ((x) > 0 ? (x) : (-x))
const int mod = 1e9 + 7;
inline int MOD(int x) {
while (x < 0) x += mod;
while (x >= mod) x -= mod;
return x;
}
namespace Newstd {
char buf[1 << 21],*p1 = buf,*p2 = buf;
inline int getc() {
return p1 == p2 && (p2 = (p1 = buf) + fread(buf,1,1 << 21,stdin),p1 == p2) ? EOF : *p1 ++;
}
inline int read() {
int ret = 0,f = 0;char ch = getc();
while (!isdigit(ch)) {
if(ch == '-') f = 1;
ch = getc();
}
while (isdigit(ch)) {
ret = (ret << 3) + (ret << 1) + ch - 48;
ch = getc();
}
return f ? -ret : ret;
}
inline void write(int x) {
if(x < 0) {
putchar('-');
x = -x;
}
if(x > 9) write(x / 10);
putchar(x % 10 + '0');
}
}
using namespace Newstd;
using namespace std;
const int ma = 1e5 + 5;
int a[ma],b[ma],f[ma],g[ma],ans[ma];
map<int,int>mp[ma];
int n;
struct BIT {
int tr[ma];
#define lowbit(x) (x & -x)
inline void update(int x,int k) {
for (;x < ma;x += lowbit(x)) {
tr[x] = max(tr[x],k);
}
}
inline int query(int x) {
int res = 0;
for (;x;x -= lowbit(x)) {
res = max(res,tr[x]);
}
return res;
}
inline int getsum(int l,int r) {
return query(r) - query(l - 1);
}
#undef lowbit
} bita,bitb;
int main(void) {
#ifndef ONLINE_JUDGE
freopen("in.txt","r",stdin);
#endif
n = read();
for (register int i = 1;i <= n; ++ i) a[i] = read();
for (register int i = 1;i <= n; ++ i) b[i] = ma - a[i];//方便求 g[i],直接将 a 序列的元素大小关系颠倒过来
int maxx = 0;
for (register int i = 1;i <= n; ++ i) {
f[i] = bita.getsum(1,a[i] - 1) + 1;
bita.update(a[i],f[i]);
maxx = max(maxx,f[i]);
}
for (register int i = n;i >= 1; -- i) {
g[i] = bitb.getsum(1,b[i] - 1) + 1;
bitb.update(b[i],g[i]);
}
for (register int i = 1;i <= n; ++ i) {
if (f[i] + g[i] == maxx + 1) {
mp[f[i]][g[i]] ++;
} else {//a[i] 不在任何最长上升子序列
ans[i] = 1;
}
}
for (register int i = 1;i <= n; ++ i) {
if (f[i] + g[i] == maxx + 1) {
if (mp[f[i]][g[i]] >= 2) {
ans[i] = 2;
} else {
ans[i] = 3;
}
}
}
for (register int i = 1;i <= n; ++ i) printf("%d",ans[i]);
return 0;
}