結果

問題 No.1825 Except One
コンテスト
ユーザー tktk_snsn
提出日時 2022-01-28 22:12:10
言語 PyPy3
(8.0.0 + ACL)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 879 ms / 3,000 ms
+ 329µs
コード長 684 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 128 ms
コンパイル使用メモリ 64,512 KB
実行使用メモリ 117,504 KB
最終ジャッジ日時 2026-10-20 03:51:52
合計ジャッジ時間 7,594 ms
ジャッジサーバーID
（参考情報） 		judge6_0 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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