結果

問題 No.1313 N言っちゃダメゲーム (4)
ユーザー umezo
提出日時 2020-12-10 00:36:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 75 ms / 2,000 ms
コード長 2,967 bytes
コンパイル時間 2,184 ms
コンパイル使用メモリ 201,452 KB
最終ジャッジ日時 2025-01-16 21:05:47
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;

#include<bits/stdc++.h>
using namespace std;

const int INF=2147483647;

template <typename X>
struct SegTree{
  using FX=function<X(X,X)>;// X•X -> X となる関数の型
  int n;
  FX fx;
  const X ex;
  vector<X> dat;
  SegTree(int n_,FX fx_,X ex_) : n(),fx(fx_),ex(ex_),dat(n_*4,ex_){
    int x=1;
    while (n_>x){
      x *= 2;
    }
    n=x;
  }
  void set(int i,X x){ dat[i+n-1]=x;}
  void build(){
    for (int k=n-2;k>=0;k--) dat[k]=fx(dat[2*k+1],dat[2*k+2]);
  }
  void update(int i,X x){
    i += n-1;
    dat[i]=x;
    while(i>0){
      i=(i-1)/2; // parent
      dat[i]=fx(dat[i*2+1],dat[i*2+2]);
    }
  }
  X get(int i){return dat[i+n-1];}
  X all(){return dat[0];}
  X query(int a,int b){ return query_sub(a,b,0,0,n);}
  X query_sub(int a,int b,int k,int l,int r){
    if(r<=a || b<=l){
      return ex;
    }else if(a<=l && r<=b){
      return dat[k];
    }else{
      X vl=query_sub(a,b,k*2+1,l,(l+r)/2);
      X vr=query_sub(a,b,k*2+2,(l+r)/2,r);
      return fx(vl,vr);
    }
  }
  int find_rightest(int a,int b,X x){ return find_rightest_sub(a,b,x,0,0,n);}
  int find_leftest(int a,int b,X x){ return find_leftest_sub(a,b,x,0,0,n);}
  int find_rightest_sub(int a,int b,X x,int k,int l,int r){
    if(dat[k]>x || r<=a || b<=l){  // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn a-1
      return a-1;
    }else if (k>=n-1){  // 自分が葉ならその位置をreturn
      return (k-(n-1));
    }else{
      int vr=find_rightest_sub(a,b,x,2*k+2,(l+r)/2,r);
      if(vr!=a-1){  // 右の部分木を見て a-1 以外ならreturn
        return vr;
      }else{  // 左の部分木を見て値をreturn
        return find_rightest_sub(a,b,x,2*k+1,l,(l+r)/2);
      }
    }
  }
  int find_leftest_sub(int a,int b,X x,int k,int l,int r){
    if(dat[k]>x || r<=a || b<=l){  // 自分の値がxより大きい or [a,b)が[l,r)の範囲外ならreturn b
      return b;
    }else if (k>=n-1){  // 自分が葉ならその位置をreturn
      return (k-(n-1));
    }else{
      int vl=find_leftest_sub(a,b,x,2*k+1,l,(l+r)/2);
      if(vl != b){  // 左の部分木を見て b 以外ならreturn
        return vl;
      }else{  // 右の部分木を見て値をreturn
        return find_leftest_sub(a,b,x,2*k+2,(l+r)/2,r);
      }
    }
  }
};

int main(){
  int n,k;
  string s;
  cin>>n>>k>>s;
  
  //fx:二項演算,ex:単位元,0-index
  using X=int;
  auto fx=[](X x1,X x2)->X{ return min(x1,x2);};
  X ex=INF;
  SegTree<X> seg(200010,fx,ex);
  rep(i,n+1) seg.set(i,1);
  seg.build();
  
  vector<int> B;
  for(int i=n-1;i>=1;i--){
    if(s[i-1]=='x') continue;
    if(seg.query(i+1,min(i+k+1,n))>0){
      seg.update(i,0);
      if(i<=k) B.push_back(i);
      continue;
    }
  }
  
  if(seg.query(1,k+1)>0) cout<<0<<endl;
  else{
    reverse(ALL(B));
    rep(i,B.size()) cout<<B[i]<<endl;
  }
      
  return 0;
}
0