[BZOJ5484] [USACO18DEC-Platinum] Sort It Out [树状数组&动态规划] [KthLIS]

[$\mathfrak{L}\mathfrak{i}\mathfrak{n}\mathfrak{k}$]

---

拼题，奶牛题经典老梗加上模板

这种有点冒泡的东西 很明显一次奶牛自走之后，其他奶牛的相对位置不变。
选最长上升子序列即可。然后要选的子集就是 LIS 的补集。

把要选的子集字典序第 k 小转化成 LIS 的字典序第 k 大。套路，用树状数组可以求。

小心溢出。

---

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<vector>
using namespace std;

const int MAXN = 1e5 + 5;
const long long MAXK = 1e18 + 5;

long long k;
int n, a[MAXN], Len;
int q[MAXN], qh = 1, qt = 0;
vector<int>Lst[MAXN];
bool Selected[MAXN];

struct Arr
{
	int Len;
	long long Rnk;
	Arr(int a = 0, long long b = 0)
	{
		Len = a;
		Rnk = b;
	}
} F[MAXN], Bit[MAXN];

void Modify(Arr &a, const Arr &b)
{
	if (a.Len > b.Len) return;
	if (a.Len < b.Len) 
	{
		a = b;
		return;
	}
	a.Rnk = min(MAXK, a.Rnk + b.Rnk);
}

void Update(int Pos, const Arr &x)
{
	while (Pos)
	{
		Modify(Bit[Pos], x);
		Pos -= Pos & -Pos;
	}
}

Arr Query(int Pos)
{
	Arr Ans = Arr(0, 1);
	while (Pos <= n)
	{
		Modify(Ans, Bit[Pos]);
		Pos += Pos & -Pos;
	}
	return Ans;
}

int main()
{
//	freopen("itout.in", "r", stdin);
//	freopen("itout.out", "w", stdout);
	scanf("%d%lld", &n, &k);
	for (int i = 1; i <= n; ++i)
	{
		scanf("%d", &a[i]);
	}
	for (int i = n; i >= 1; --i)
	{
		F[i] = Query(a[i] + 1);
		Lst[++F[i].Len].push_back(i);
		Update(a[i], F[i]);
	}
	int LISLen = Query(1).Len;
	
	if (n - LISLen < k)
	{
		;
	}
	
	for (int ArrNow = 0, Cur = LISLen; Cur; --Cur)
	{
		for (int tp, i = Lst[Cur].size() - 1; i >= 0; --i)
		{
			if (F[tp = Lst[Cur][i]].Rnk < k)
			{
				k -= F[tp].Rnk;
			}
			else
			{
				Selected[a[tp]] = 1;
				while (ArrNow < tp)
				{
					F[++ArrNow].Rnk = 0;
				}
				break;
			}
		}
	}
	printf("%d\n",n - LISLen);
	for (int i = 1; i <= n; ++i)
	{
		if (!Selected[i])
		{
			printf("%d\n", i);
		}
	}
	return 0;
}