题目链接
题意:给你长度 n + 1 n+1 n+1的一个 01 01 01串,下标 [ 0 , n + 1 ] [0,n+1] [0,n+1] , 你从 0 0 0出发,每次走的步数 的范围 [ 1 , m ] [1,m] [1,m],让你走到 n + 1 n+1 n+1 0 0 0的地方可以走, 1 1 1的地方不能走,使得步数最少的情况下,方案的字典序最小。输出这个方案。无解输出 1 -1 1

我们先用线段树维护出每个 0 0 0走到终点最少多少步记为数组 f f f,然后从起点开始遍历,因为有解的情况下, f 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;
}