結果

問題 No.1021 Children in Classrooms
ユーザー eQe
提出日時 2020-04-29 01:43:24
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 90 ms / 2,000 ms
コード長 2,349 bytes
コンパイル時間 1,464 ms
コンパイル使用メモリ 163,568 KB
実行使用メモリ 7,228 KB
最終ジャッジ日時 2024-11-26 09:21:17
合計ジャッジ時間 3,378 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <string>
#define ft first
#define sc second
#define pt(sth) cout << sth << "\n"
#define chmax(a, b) (a)=max(a, b)
#define chmin(a, b) (a)=min(a, b)
#define moC(a, s, b) (a)=((a)s(b)+MOD)%MOD
using namespace std;
typedef long long ll;
typedef pair<ll, ll> pll;
static const ll INF=1e18;
static const ll MAX=101010;
static const ll MOD=1e9+7;

/*
 for(i=0; i<N; i++)
   cin >> a[i];
*/

ll N, M;
string s;

ll collision_left(ll k) {
  if(k==0) return 1;
  
  ll i;
  for(i=0; i<M; i++) {
    if(s[i]=='L') k--;
    else k++;
    
    if(k==0) return 1;
    
    if(k>=N) k=N-1;
  }
  
  return 0;
}

ll collision_right(ll k) {
  if(k==N-1) return 1;
  
  ll i;
  for(i=0; i<M; i++) {
    if(s[i]=='L') k--;
    else k++;
    
    if(k==N-1) return 1;
    else if(k<0) k=0;
  }
  
  return 0;
}

int main(void) {
  ll i, j, k;
  
  cin >> N >> M;
  ll a[MAX*2];
  
  for(i=0; i<N; i++) cin >> a[i];
  cin >> s;
  
  
  ll l=0, r=N;
  while(l+1<r) {
    ll m=(l+r)/2;
    (collision_left(m) ? l:r)=m;
  }
  ll L=l;
  
  l=0; r=N;
  while(l+1<r) {
    ll m=(l+r)/2;
    (collision_right(m) ? r:l)=m;
  }
  if(collision_right(0)) r--;
  ll R=r;
  
  
  if(L>=R) {
    ll sum=0;
    for(i=0; i<N; i++) sum+=a[i];
    
    k=0;
    for(i=0; i<M; i++) {
      if(s[i]=='L') k--;
      else k++;
      
      if(k<0) k=0;
      else if(k>=N) k=N-1;
    }
    
    
    
    vector<ll> v;
    for(i=0; i<N; i++) {
      if(i==k) v.push_back(sum);
      else v.push_back(0);
    }
    
    for(i=0; i<N; i++) {
      if(i==0) cout << v[i];
      else cout << " " << v[i];
    }
    pt("");
    
  }else {
    ll sl=0, sr=0;
    for(i=0; i<=L; i++) sl+=a[i];
    for(i=R; i<N; i++) sr+=a[i];
    
    k=0;
    for(i=0; i<M; i++) {
      if(s[i]=='L') k--;
      else k++;
      
      if(k<0) k=0;
      else if(k>=N) k=N-1;
    }
    ll lk=k;
    
    k=N-1;
    for(i=0; i<M; i++) {
      if(s[i]=='L') k--;
      else k++;
      
      if(k<0) k=0;
      else if(k>=N) k=N-1;
    }
    ll rk=k;
    
    vector<ll> v;
    for(i=0; i<lk; i++) v.push_back(0);
    v.push_back(sl);
    for(i=lk+1, j=1; i<rk; i++, j++) v.push_back(a[L+j]);
    v.push_back(sr);
    for(i=rk+1; i<N; i++) v.push_back(0);
    
    for(i=0; i<N; i++) {
      if(i==0) cout << v[i];
      else cout << " " << v[i];
    }
    pt("");
  }
  
  
}
0