AtCoder Beginner Contest 146 F - Sugoroku(线段树+思维)

题目链接
题意：给你长度$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;
}