题目链接
题意:给你长度 n+1的一个 01串,下标 [0,n+1] , 你从 0出发,每次走的步数 的范围 [1,m],让你走到 n+1, 0的地方可以走, 1的地方不能走,使得步数最少的情况下,方案的字典序最小。输出这个方案。无解输出 −1
我们先用线段树维护出每个 0走到终点最少多少步记为数组 f,然后从起点开始遍历,因为有解的情况下, f数组必然是递减的,那么直接遍历串,第一次开始减的地方就是我下一步要到的地方。这样就能保证字典序最小了。
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 2e5 + 10;
#define fi first
#define se second
#define pb push_back
int n,k;
char a[N];
int b[N];
#define mid (l+r>>1)
#define ls o<<1
#define rs o<<1|1
int t[N<<2];
void build(int o,int l,int r){
if(l==r){
t[o]=b[l];
return ;
}
t[o]=0;
build(ls,l,mid);
build(rs,mid+1,r);
t[o]=max(t[ls],t[rs]);
}
const int MAXN=1e9;
int get(int o,int l,int r,int x,int y){
if(l>y||r<x||l>r)return 0;
if(l>=x&&r<=y)return t[o];
int ans=0;
if(y>mid&&t[rs])ans=max(ans,get(rs,mid+1,r,x,y));
if(x<=mid&&ans==0&&t[ls])ans=max(ans,get(ls,l,mid,x,y));
return ans;
}
int f[N];
int main() {
ios::sync_with_stdio(false);
cin>>n>>k>>a+1;
n++;
for(int i=1;i<=n;i++){
if(a[i]=='1')b[i]=0;
else b[i]=i;
}
build(1,1,n);
for(int i=n-1;i>=1;i--){
if(a[i]=='0'){
int nex=get(1,1,n,i+1,min(n,i+k));
if(nex==0){
f[i]=MAXN;
continue;
}
f[i]=f[nex]+1;
}else f[i]=MAXN;
}
if(f[1]>=MAXN)cout<<-1<<endl;
else{
int l=2;
int pre=1;
while(l<=n){
if(n>pre&&n-pre<=k){
return cout<<n-pre<<' ',0;
}
if(a[l]=='1'||f[l]>=f[pre]){
l++;
continue;
}
cout<<l-pre<<' ';
pre=l;
l++;
}
}
return 0;
}