結果
問題 |
No.1433 Two color sequence
|
ユーザー |
|
提出日時 | 2021-08-29 01:00:58 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,161 bytes |
コンパイル時間 | 928 ms |
コンパイル使用メモリ | 82,156 KB |
実行使用メモリ | 156,868 KB |
最終ジャッジ日時 | 2024-11-22 01:14:47 |
合計ジャッジ時間 | 6,659 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 WA * 9 |
ソースコード
#!/usr/bin/env python3 import sys def main(): import itertools N = int(input()) S = input() A = list(map(int, input().split())) RmB = [] mRB = [] for i in range(N): if S[i] == "B": RmB.append(-A[i]) mRB.append(A[i]) else: RmB.append(A[i]) mRB.append(-A[i]) sRmB = [0] + list(itertools.accumulate(RmB)) smRB = [0] + list(itertools.accumulate(mRB)) # print(RmB) # print(mRB) # print(sRmB) # print(smRB) minS = 10 ** 10 maxS = -1 * 10 ** 10 ans = -1 * 10 ** 10 for i in range(N + 1): # print(sRmB[i], minS) if sRmB[i] <= minS: ans = max(ans, maxS - minS) minS = sRmB[i] maxS = -1 * 10 ** 10 maxS = max(sRmB[i], maxS) ans = max(ans, maxS - minS) for i in range(N + 1): # print(smRB[i], minS) if smRB[i] <= minS: ans = max(ans, maxS - minS) minS = smRB[i] maxS = -1 * 10 ** 10 maxS = max(smRB[i], maxS) ans = max(ans, maxS - minS) print(ans) return if __name__ == '__main__': main()