洛谷5156 Sort It Out

Problem

洛谷

Solution

充分必要条件就是未被选择的数是单调上升的，要求最少，则需要求LIS。而被选择的字典序第k小也就对应着被选择的字典序第k大。
那么我们考虑从高位到低位确定下来每一个数字，由此我们需要求出选了某个数之后的后继LIS方案数，用树状数组优化dp解决。
然后我们就按顺序丢进vector里即可，注意其实这里vector内部是不需要排序的，因为我们是从前往后扫的，若同样长度的vector存在一个后进入的数值比前面的数值大，则显然它的LIS应该更长，矛盾。
时间复杂度$O(n\log n)$

Code

#include <assert.h>
#include <cstdio>
#include <vector>
#define lowbit(x) ((x)&(-(x)))
using namespace std;
typedef long long ll;
const int maxn=100010;
const ll INF=1e18;
template <typename Tp> inline int getmin(Tp &x,Tp y){return y<x?x=y,1:0;}
template <typename Tp> inline int getmax(Tp &x,Tp y){return y>x?x=y,1:0;}
template <typename Tp> inline void read(Tp &x)
{
    x=0;int f=0;char ch=getchar();
    while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
    if(ch=='-') f=1,ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    if(f) x=-x;
}
int n,mx,now,val[maxn],vis[maxn];
ll k;
struct data{
	int len;ll sum;
	data(const int _len=0,const ll _sum=0ll){len=_len;sum=_sum;}
	data operator + (const data &b)const
	{
		data res;
		if(len>b.len) res=*this;
		if(len<b.len) res=b;
		if(len==b.len)
		{
			res.len=len;
			res.sum=sum+b.sum;
			if(res.sum>INF) res.sum=INF;
		}
		return res;
	}
	void operator += (const data &b){*this=*this+b;}
}a[maxn],f[maxn];
vector<data> G[maxn];
vector<data>::iterator itr;
void add(int p,data v){for(++p;p<=n+1;p+=lowbit(p)) a[p]+=v;}
data query(int p)
{
	data res(0,0);
	for(++p;p;p-=lowbit(p)) res+=a[p];
	return res;
}
int main()
{
	read(n);read(k);
	add(0,data(0,1));
	for(int i=1;i<=n;i++) read(val[i]);
	for(int i=n;i;i--)
	{
		f[i]=query(n-val[i]+1);
		++f[i].len;getmax(mx,f[i].len);
		add(n-val[i]+1,f[i]);
	}
	for(int i=1;i<=n;i++) G[f[i].len].push_back(data(i,f[i].sum));
	for(int i=mx;i;i--)
	{
		itr=G[i].begin();
		while(itr!=G[i].end()&&itr->len<now) ++itr;
		assert(itr!=G[i].end());
		while(itr->sum<k) k-=itr->sum,++itr;
		now=itr->len;vis[val[now]]=1;
	}
	printf("%d\n",n-mx);
	for(int i=1;i<=n;i++) if(!vis[i]) printf("%d\n",i);
	return 0;
}