結果
問題 | No.3017 交互浴 |
ユーザー |
|
提出日時 | 2025-01-25 13:14:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 196 ms / 2,000 ms |
コード長 | 603 bytes |
コンパイル時間 | 714 ms |
コンパイル使用メモリ | 81,536 KB |
実行使用メモリ | 107,008 KB |
最終ジャッジ日時 | 2025-01-25 22:39:46 |
合計ジャッジ時間 | 16,175 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 55 |
ソースコード
import sys input = lambda: sys.stdin.readline().rstrip() def main(): # 入力 N = int(input()) H = list(map(int, input().split())) # 計算・出力 stack = [(10**10, -1)] # 色の変わり目と色 0:そこまで緑, 1:そこまで水 ans = 0 for i, h in enumerate(H): c = -1 if (i & 1) else 1 while stack[-1][0] <= h: ph, pc = stack.pop() ans -= ph * pc if stack[-1][1] != c: stack.append((h, c)) ans += h * c print(ans) # print(f'{stack = }') if __name__ == "__main__": main()