結果

問題 No.1021 Children in Classrooms
ユーザー kimiyuki
提出日時 2020-04-10 22:05:38
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 344 ms / 2,000 ms
コード長 1,176 bytes
コンパイル時間 120 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 30,300 KB
最終ジャッジ日時 2024-09-15 20:27:09
合計ジャッジ時間 5,089 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

#!/usr/bin/env python3
from typing import *

def solve(n: int, m: int, a: List[int], s: str) -> Any:
    l = 0
    r = n
    cnt_l = 0
    cnt_r = 0
    for c in s:
        if c == 'L':
            if l == 0:
                cnt_l += 1
            else:
                l -= 1
            r = max(1, r - 1)
        elif c == 'R':
            l = min(n - 1, l + 1)
            if r == n:
                cnt_r += 1
            else:
                r += 1
        else:
            assert False
    b = [0] * n
    for i in range(n):
        if i < cnt_l:
            b[l] += a[i]
        elif cnt_l <= i and i < n - cnt_r:
            b[l + i - cnt_l] += a[i]
        else:
            b[r - 1] += a[i]
    return b

# generated by online-judge-template-generator (https://github.com/kmyk/online-judge-template-generator)
def main():
    import sys
    tokens = iter(sys.stdin.read().split())
    N = int(next(tokens))
    M = int(next(tokens))
    a = [None for _ in range(N)]
    for i in range(N):
        a[i] = int(next(tokens))
    S = next(tokens)
    assert next(tokens, None) is None
    ans = solve(N, M, a, S)
    print(*ans)

if __name__ == '__main__':
    main()
0