結果

問題 No.973 余興
ユーザー beetbeet
提出日時 2020-01-17 21:56:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,916 ms / 4,000 ms
コード長 2,167 bytes
コンパイル時間 1,887 ms
コンパイル使用メモリ 199,668 KB
最終ジャッジ日時 2025-01-08 18:30:42
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 54
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}


template<typename T>
struct BIT{
  int n;
  vector<T> bit;
  //1-indexed
  BIT(int n_):n(n_+1),bit(n+1,0){}

  T sum(int i){
    T s(0);
    for(int x=i;x>0;x-=(x&-x))
      s+=bit[x];
    return s;
  }

  void add(int i,T a){
    if(i==0) return;
    for(int x=i;x<=n;x+=(x&-x))
      bit[x]+=a;
  }

  int lower_bound(int w){
    if(w<=0) return 0;
    int x=0,r=1;
    while(r<n) r<<=1;
    for(int k=r;k>0;k>>=1){
      if(x+k<=n&&bit[x+k]<w){
        w-=bit[x+k];
        x+=k;
      }
    }
    return x+1;
  }

  // [l, r)
  T query(int l,int r){
    return sum(r-1)-sum(l-1);
  }

  T sum0(int i){
    return sum(i+1);
  }

  void add0(int i,T a){
    add(i+1,a);
  }

  T query0(int l,int r){
    return sum(r)-sum(l);
  }
};

//INSERT ABOVE HERE
const int MAX = 5050;
int dp[MAX][MAX]={};

signed main(){
  int n,x;
  cin>>n>>x;
  vector<int> as(n);
  for(int i=0;i<n;i++) cin>>as[i];

  //vector<int> sm(n+1,0);
  //for(int i=0;i<n;i++) sm[i+1]=sm[i]+as[i];

  vector<int> ls(n),rs(n);
  for(int i=0;i<n;i++){
    {
      int j=i-1;
      long long sm=0;
      while(j>=0&&sm+as[j]<=x){
        sm+=as[j];
        j--;
      }
      ls[i]=j+1;
    }
    {
      int j=i+1;
      long long sm=0;
      while(j<n&&sm+as[j]<=x){
        sm+=as[j];
        j++;
      }
      rs[i]=j-1;
    }
    // cout<<i<<":"<<ls[i]<<" "<<rs[i]<<endl;
  }

  using B = BIT<int>;
  vector<B> bl(n,B(n+1));
  vector<B> br(n,B(n+1));
  for(int w=1;w<=n;w++){
    for(int l=0;l+w<=n;l++){
      // [l, r]
      int r=l+w-1;
      dp[l][r]|=bl[l].sum0(r);
      dp[l][r]|=br[r].sum0(l);
      if(dp[l][r]) continue;

      // cout<<l<<" "<<r<<endl;
      if(r+1<n){
        // [r+1, rs[r]]
        bl[l].add0(r+1,1);
        bl[l].add0(rs[r]+1,-1);
      }
      // cout<<l<<" "<<r<<endl;
      if(l>0){
        // [ls[l], l-1]
        br[r].add0(ls[l],1);
        br[r].add0(l,-1);
      }
    }
  }

  cout<<(dp[0][n-1]?"A":"B")<<endl;
  return 0;
}
0