虽说是道绿题,但个人感觉思路不太好想
首先,我们假设可以路过已消除的砖块,并考虑用栈维护
我们将元素(砖块)按顺序压入栈,每当压入一个元素,就检查栈顶
k
+
1
k+1
k+1 个元素中是否恰好有一块黑色的砖。如果满足此条件,那么就将这
k
+
1
k+1
k+1 块消掉。
把这个方案画成图就是这个样子
可以发现,这些不同颜色的弧是互相包含的,并且可以证明不会两弧交叉
那么,我们把上面方案的操作顺序倒过来,满足先做没有被包含的操作。这样就不会经过已经被消除的块了。
这里感性理解一下为什么两弧不会交叉
显然,在进行红色操作时,
B
\text{B}
B 块肯定比
A
\text{A}
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;
}