洛谷 P3551 [POI2013]USU-Take-out 题解

题目链接

虽说是道绿题，但个人感觉思路不太好想

首先，我们假设可以路过已消除的砖块，并考虑用栈维护
我们将元素（砖块）按顺序压入栈，每当压入一个元素，就检查栈顶 $k+1$ 个元素中是否恰好有一块黑色的砖。如果满足此条件，那么就将这 $k+1$ 块消掉。

把这个方案画成图就是这个样子

可以发现，这些不同颜色的弧是互相包含的，并且可以证明不会两弧交叉
那么，我们把上面方案的操作顺序倒过来，满足先做没有被包含的操作。这样就不会经过已经被消除的块了。

这里感性理解一下为什么两弧不会交叉

显然，在进行红色操作时，$\text{B}$ 块肯定比 $\text{A}$ 块优先选，所以一定不会出现交叉的情况

#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
const int Maxn=1000000+10;
vector <int> g[Maxn];
int a[Maxn],c[Maxn];
int sum[Maxn];
char s[Maxn];
int n,k,cnt,tot;
inline int lowbit(int x)
{
    
    
	return x&(-x);
}
void modify(int x,int k)
{
    
    
	while(x<=n)
	{
    
    
		sum[x]+=k;
		x+=lowbit(x);
	}
}
int query(int x)
{
    
    
	int ret=0;
	while(x)
	{
    
    
		ret+=sum[x];
		x-=lowbit(x);
	}
	return ret;
}
int main()
{
    
    
//	freopen("in.txt","r",stdin);
	scanf("%d%d",&n,&k);
	scanf("%s",s+1);
	for(int i=1;i<=n;++i)
	if(s[i]=='c')a[i]=1;
	else a[i]=0;
	for(int i=1;i<=n;++i)
	{
    
    
		c[++cnt]=i;
		if(s[i]=='c')modify(cnt,1);
		if(cnt>k && query(cnt)-query(cnt-k-1)==1)
		{
    
    
			++tot;
			for(int j=1;j<=k+1;++j)
			{
    
    
				if(s[c[cnt]]=='c')
				modify(cnt,-1);
				g[tot].push_back(c[cnt--]);
			}
		}
	}
	for(int i=tot;i;--i)
	{
    
    
		for(int j=g[i].size()-1;j>=0;--j)
		printf("%d ",g[i][j]);
		putchar('\n');
	}
	return 0;
}