牛客挑战赛36-C 纸飞机 //dp优化

题面链接

https://ac.nowcoder.com/acm/contest/3782/C

题目大意

对于原数组的每个数$a[i]$，分别求出去除$a[i]$后，能覆盖另外$n-1$个数的最少递减子序列的个数。

思路

首先，去除$a[i]$后，需要用最少的递减子序列覆盖另外$n-1$个数，也就是求：去除$a[i]$后的最小链覆盖。根据结论最小链覆盖等于最长反链，也就是求：去除$a[i]$后，序列的最长不下降子序列。

对于原序列的最长非下降子序列，很容易求解：定义$dp[i]$为以$a[i]$结尾的最长不下降子序列长度，状态转移方程为：$dp[i]=max(dp[j])+1(a[j]<=a[i]且j<i)$，由于$n$范围较大，可以用树状数组/线段树维护前缀最大值，实现$O(nlo{g}_n)$复杂度求解原数组每个数的最长不下降子序列长度，由于$a[i]$范围较大，需要先对原数组进行离散化，才好用数据结构维护。

设求出的原序列最长不下降子序列长度为$ans$，对于每个数$a[i]$，去除之后，序列的最长不下降子序列长度和"$a[i]$是否为原序列最长不下降子序列的一员且无法被取代"有关。如果是且无法被取代，则结果为$ans-1$；否则为$ans$。

用一个辅助数组$ma$，定义$ma[i]$为放在原序列最长非递减子序列第$i$位的最大值，将$ma[ans+1]$设为无穷大，方便统一处理。再用一个$vis$数组，记录第$i$个数是否可以成为原序列最长非递减子序列的一员。然后从后往前维护这两个数组：如果$ma[dp[i]+1]>=a[i]$，说明$a[i]$可以是原序列最长非递减子序列的第$i$位，也就是$a[i]$可以成为原序列最长非递减子序列的一员，但是否为原序列最长非递减子序列第$i$位的最大值需要判断一下来更新维护。

之后，再用一个$sum$数组，来记录有多少个数可以是原序列最长非递减子序列的第$i$位，目的是：判断$a[i]$能否被取代，当$vis[i]且sum[dp[i]]=1$时，说明$a[i]$可以是原序列最长非递减子序列的一员，且不能被取代。否则说明可以被取代。

参考代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <map>
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const int maxn=1e6+7;
const int inf=0x3f3f3f3f;
#define pb push_back
#define ft first
#define sd second
#define ms(x,y) memset(x,y,sizeof(x))
int n,ans;
int a[maxn],b[maxn],dp[maxn];
int ma[maxn],sum[maxn];
bool vis[maxn];
struct BIT{
    int bit[maxn];
    inline int lowbit(int x){return x&(-x);}
    inline void init(){ for(int i=1;i<=n;i++) bit[i]=0; }
    inline void update(int x,int v){for(int i=x;i<=n;i+=lowbit(i))bit[i]=max(bit[i],v);}
    inline int query(int x,int ans=0){for(int i=x;i;i-=lowbit(i))ans=max(bit[i],ans);return ans;}
}s;
void lis(){//离散化
    sort(b+1,b+1+n);
    int len=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=n;i++){
        a[i]=lower_bound(b+1,b+1+len,a[i])-b;
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    lis();
    s.init();
    for(int i=1;i<=n;i++){
        dp[i]=s.query(a[i])+1;
        s.update(a[i],dp[i]);
        ans=max(ans,dp[i]);
    }
    ma[ans+1]=inf;
    for(int i=n;i>=1;i--){
        if(ma[dp[i]+1]>=a[i]){//a[i]能否放在原序列最长非下降子序列第dp[i]位
            ma[dp[i]]=max(ma[dp[i]],a[i]);//a[i]能否成为原序列最长非下降子序列第dp[i]位的最大值
            vis[i]=true;//标记一下
        }
    }
    for(int i=1;i<=n;i++)if(vis[i])sum[dp[i]]++;//统计有多少个数可以是原序列最长非递减子序列的第$i$位
    for(int i=1;i<=n;i++){
        printf("%d%c",(vis[i]&&sum[dp[i]]==1)?ans-1:ans,i==n?'\n':' ');
    }
    return 0;
}