結果
問題 |
No.258 回転寿司(2)
|
ユーザー |
|
提出日時 | 2025-02-14 01:16:50 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 66 ms / 2,000 ms |
コード長 | 1,171 bytes |
コンパイル時間 | 472 ms |
コンパイル使用メモリ | 82,040 KB |
実行使用メモリ | 66,220 KB |
最終ジャッジ日時 | 2025-02-14 01:16:58 |
合計ジャッジ時間 | 7,738 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 67 |
ソースコード
import sys import math import bisect from heapq import heapify, heappop, heappush from collections import deque, defaultdict, Counter from functools import lru_cache from itertools import accumulate, combinations, permutations, product sys.set_int_max_str_digits(10 ** 6) sys.setrecursionlimit(1000000) MOD = 10 ** 9 + 7 MOD99 = 998244353 input = lambda: sys.stdin.readline().strip() NI = lambda: int(input()) NMI = lambda: map(int, input().split()) NLI = lambda: list(NMI()) SI = lambda: input() SMI = lambda: input().split() SLI = lambda: list(SMI()) EI = lambda m: [NLI() for _ in range(m)] def main(): N = NI() V = NLI() dp = [[[0, -1] for _ in range(2)] for _ in range(N+1)] for i in range(N): v = V[i] dp[i+1][0] = dp[i][0][:] if dp[i+1][0][0] < dp[i][1][0]: dp[i+1][0] = dp[i][1][:] dp[i+1][1] = [dp[i][0][0]+v, i] # print(*dp, sep="\n") if dp[N][0][0] > dp[N][1][0]: v, t = dp[N][0] else: v, t = dp[N][1] print(v) ans = [] while t >= 0: ans.append(t+1) v, t = dp[t][0] print(*ans[::-1]) if __name__ == "__main__": main()