P3551 [POI2013] USU-Take-out 题解

考虑这样一个做法：

从前往后将每一块砖推入栈顶，如果栈顶连续 $k+1$ 个元素有 $k$ 个白色，$1$ 个黑色，就把这些捆绑作为一组。最后逆序输出这些组。

显然这是正确的，因为这不会导致经过消除过的砖。

考虑使用平衡树维护，开两个平衡树，分别维护黑色和白色的栈，每次相当于问黑色的第 $k$ 大的与白色第二大的相对关系即可。

为了容易实现，考虑用 pb_ds 维护即可。

#include <bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
const int N = 1e6 + 5;

vector<vector<int>> v;

int n, k;
string s;

tree<int, null_type, greater<int>, rb_tree_tag, tree_order_statistics_node_update> s1, s2;

int main()
{
	ios::sync_with_stdio(0), cin.tie(0);
	cin >> n >> k >> s;
	s = " " + s;
	for (int i = 1; i <= n; i++)
	{
		if (s[i] == 'c') s2.insert(i);
		else s1.insert(i);
		if (s2.size() && s1.size() >= k)
		{
			if (s2.size() >= 2 && *s1.find_by_order(k - 1) < *s2.find_by_order(1)) continue;
			vector<int> g;
			int ggg = *s2.find_by_order(0);
			g.emplace_back(ggg);
			s2.erase(ggg);
			int c = k;
			vector<int> kk; 
			while (c--)
			{
				g.emplace_back(*s1.find_by_order(0));
				kk.emplace_back(*s1.find_by_order(0));
				s1.erase(s1.find_by_order(0));
			}
			sort(g.begin(), g.end());
			v.emplace_back(g);
		}
	}
	reverse(v.begin(), v.end());
	for (auto &i : v)
	{
		for (auto &j : i) cout << j << " ";
		cout << "\n";
	}
	return 0;
}