結果

問題 No.1021 Children in Classrooms
コンテスト
ユーザー ei13337
提出日時 2020-04-10 22:34:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,973 bytes
コンパイル時間 1,311 ms
コンパイル使用メモリ 112,880 KB
実行使用メモリ 11,212 KB
最終ジャッジ日時 2024-09-15 21:08:23
合計ジャッジ時間 2,832 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 10 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <stack>
#include <queue>
#include <algorithm>
#include <cassert>
#include <random>
#include <iomanip>
#include <bitset>
#define FOR(i, n, m) for(ll i = n; i < (int)m; i++)
#define REP(i, n) FOR(i, 0, n)
#define ALL(v) v.begin(), v.end()
#define pb push_back
using namespace std;
using ll = long long;
using P = pair<ll, ll>;
constexpr ll inf = 1000000000;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-9;

int main() {
    cin.tie(0);
    ios::sync_with_stdio(false);
    ll n, m;
    cin >> n >> m;
    vector<ll> a(n);
    REP(i, n) cin >> a[i];
    string s;
    cin >> s;
    vector<ll> l, r;
    l.pb(1); r.pb(n);
    ll lp = 1, rp = n;
    REP(i, m) {
        if(lp == rp) {
            if(s[i] == 'L') {
                if(lp != 1) {
                    lp--;
                    rp--;
                }
            } else {
                if(lp != n) {
                    lp++;
                    rp++;
                }
            }
            continue;
        }
        if(s[i] == 'L') {
            if(lp == 1) {
                l.pb(l.back() + 1);
                rp--;
            } else {
                lp--;
                rp--;
            }
        } else {
            if(rp == n) {
                r.pb(r.back() - 1);
                lp++;
            } else {
                lp++;
                rp++;
            }
        }
    }
    vector<ll> ans(n, 0);
    if(lp == rp) REP(i, r.size()) l.pb(r[i]);
    for(ll i = lp; i <= rp; i++) {
        if(i == lp) {
            REP(j, l.size()) ans[i - 1] += a[l[j] - 1];
        } else if(i == rp) {
            REP(j, r.size()) ans[i - 1] += a[r[j] - 1];
        } else {
            ans[i - 1] = a[(int)l.size() + (i - lp) - 1];
        }
    }
    REP(i, n) {
        if(i) cout << " ";
        cout << ans[i];
    }
    cout << endl;
    return 0;
}
0