結果
問題 |
No.1825 Except One
|
ユーザー |
![]() |
提出日時 | 2022-01-28 22:12:10 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 972 ms / 3,000 ms |
コード長 | 684 bytes |
コンパイル時間 | 293 ms |
コンパイル使用メモリ | 82,448 KB |
実行使用メモリ | 169,892 KB |
最終ジャッジ日時 | 2025-01-01 23:12:06 |
合計ジャッジ時間 | 6,800 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 31 |
ソースコード
from collections import defaultdict N = int(input()) A = list(map(int, input().split())) """ * sum(X) は (M-1)の倍数 * 操作回数がsum(X)//(M-1) * max(X)が操作回数以下であればいける dp[i,j,k]: 要素数i, 合計がj, maxがk -> 何通り? 状態数少なそう """ dp = [defaultdict(int) for _ in range(N+1)] dp[0][(0, 0)] = 1 for m, a in enumerate(A): for i in reversed(range(m+1)): for (j, k), v in dp[i].items(): dp[i+1][(j+a, max(k, a))] += v ans = 0 for i in range(2, N+1): for (sum, max), v in dp[i].items(): if sum % (i - 1) != 0: continue if max <= sum // (i - 1): ans += v print(ans)