[BZOJ]5484: [Usaco2018 Dec]Sort It Out 最长上升子序列

Description

FJ有N（1≤N≤10^5）头奶牛（分别用1…N编号）排成一行。FJ喜欢他的奶牛以升序排列，不幸的是现在她们的顺序被打乱了。在过去FJ曾经使用一些诸如“冒泡排序”的开创性的算法来使他的奶牛排好序，但今天他想偷个懒。取而代之，他会每次对着一头奶牛叫道“按顺序排好”。当一头奶牛被叫到的时候，她会确保自己在队伍中的顺序是正确的（从她的角度看来）。只要有一头紧接在她右边的奶牛的编号比她小，她们就交换位置。然后，只要有一头紧接在她左边的奶牛的编号比她大，她们就交换位置。这样这头奶牛就完成了“按顺序排好”，在这头奶牛看来左边的奶牛编号比她小，右边的奶牛编号比她大。FJ想要选出这些奶牛的一个子集，然后遍历这个子集，依次对着每一头奶牛发号施令（按编号递增的顺序），重复这样直到所有N头奶牛排好顺序。例如，如果她选出了编号为{2,4,5}的奶牛的子集，那么他会喊叫奶牛2，然后是奶牛4，然后是奶牛5。如果N头奶牛此时仍未排好顺序他会再次对着这几头奶牛喊叫，如果有必要的话继续重复。由于FJ不确定哪些奶牛比较专心，他想要使得这个子集最小。此外，他认为K是个幸运数字。请帮他求出满足重复喊叫可以使得所有奶牛排好顺序的最小子集之中字典序第K小的子集。我们称{1,…,N}的一个子集S在字典序下小于子集T，当S的所有元素组成的序列（按升序排列）在字典序下小于T的所有元素组成的序列（按升序排列）。例如，{1,3,6} 在字典序下小于{1,4,5}。

Solution

首先，第一问的答案为$n-$最长上升子序列长度，感性理解一下，没有在集合中的数相对顺序是不会变的，而其他都会不断调整，直到回到自己的位置。
而且，总方案数就是最长上升子序列的方案数，而且是一一对应的。要找第$K$小，就是找最长上升子序列的第$K$大，拿个树状数组乱搞就行了。

Code

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=100010;
const LL inf=2000000000000000000LL;
LL read()
{
    LL x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int a[Maxn],f[Maxn],F[Maxn],n;LL k,g[Maxn],G[Maxn];
int mx[Maxn];LL s[Maxn];
void modify(int x,int v,LL c)
{
    bool flag=false;
    for(;x<=n;x+=(x&-x))
    if(v>mx[x])mx[x]=v,s[x]=c;
    else if(v==mx[x])s[x]=min(s[x]+c,inf);
}
int V;LL C;
void query(int x)
{
    V=0;C=1;
    for(;x;x-=(x&-x))
    if(mx[x]>V)V=mx[x],C=s[x];
    else if(mx[x]==V)C=min(C+s[x],inf);
}
vector<int>h[Maxn];bool mark[Maxn];int pos[Maxn];
int main()
{
    n=read(),k=read();
    for(int i=1;i<=n;i++)a[i]=read(),pos[a[i]]=i;
    int ans=0;
     
    modify(n-a[n]+1,1,1);f[n]=g[n]=1;
    for(int i=n-1;i;i--)
    {
        sb=0;query(n-a[i]+1);f[i]=V+1,g[i]=C;
        ans=max(ans,f[i]);
        modify(n-a[i]+1,f[i],g[i]);
    }
    memset(s,0,sizeof(s));memset(mx,0,sizeof(mx));
    modify(a[1],1,1);F[1]=G[1]=1;
    for(int i=2;i<=n;i++)
    {
        query(a[i]);F[i]=V+1,G[i]=C;
        modify(a[i],F[i],G[i]);
    }
    printf("%d\n",n-ans);
     
    for(int i=1;i<=n;i++)
    if(f[i]+F[i]-1==ans)h[F[i]].push_back(a[i]);
    for(int i=1;i<=ans;i++)sort(h[i].begin(),h[i].end());
    int lst=-1,lp=-1;
    for(int i=1;i<=ans;i++)
    {
        for(int j=h[i].size()-1;j>=0;j--)
        {
            int x=h[i][j];
            if(x<lst||pos[x]<lp)continue;
            if(k<=g[pos[x]]){lst=x;lp=pos[x];mark[x]=true;break;}
            k-=g[pos[x]];
        }
    }
    for(int i=1;i<=n;i++)if(!mark[i])printf("%d\n",i);
}