結果
問題 | No.1765 While Shining |
ユーザー |
|
提出日時 | 2021-11-26 22:44:37 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 230 ms / 2,000 ms |
コード長 | 658 bytes |
コンパイル時間 | 131 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 19,772 KB |
最終ジャッジ日時 | 2024-06-29 18:18:44 |
合計ジャッジ時間 | 4,582 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
n = int(input()) A = list(map(int, input().split())) L = [] seq_num = 1 curr = A[0] for i in range(1, n): if A[i - 1] != A[i]: seq_num += 1 else: L.append((seq_num, curr)) seq_num = 1 curr = A[i] if i == n - 1: L.append((seq_num, curr)) ans = 0 l = len(L) for i in range(l): seq_num, f = L[i] if f == 0: seq_num -= 1 if i == l - 1: seq_num -= 1 if seq_num <= 0: continue if seq_num % 2 == 0: num = seq_num // 2 ans += num * (num + 1) else: num = (seq_num + 1) // 2 ans += num * num print(ans)