結果

問題 No.1825 Except One
ユーザー tktk_snsn
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0