結果

問題 No.1318 ABCD quadruplets
コンテスト
ユーザー tktk_snsn
提出日時 2020-12-25 22:41:35
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 711 bytes
コンパイル時間 171 ms
コンパイル使用メモリ 82,220 KB
実行使用メモリ 218,016 KB
最終ジャッジ日時 2024-09-22 16:05:10
合計ジャッジ時間 7,351 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 TLE * 1 -- * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
N, M = map(int, input().split())

cnt = [defaultdict(int) for _ in range(2 * M + 1)]
for a in range(M + 1):
    for b in range(M + 1):
        x = a * a + a * b + b * b
        if x <= N:
            cnt[a + b][x] += 1

cntcnt = [defaultdict(int) for _ in range(3 * M + 1)]
for ab, dd in enumerate(cnt):
    for c in range(M + 1):
        x = ab*c + c*c
        for k, v in dd.items():
            if x + k <= N:
                cntcnt[ab + c][x + k] += v

ans = [0] * (N + 1)
for abc, dd in enumerate(cntcnt):
    for d in range(M + 1):
        x = abc*d + d*d
        for k, v in dd.items():
            if x + k <= N:
                ans[x + k] += v

print(*ans, sep="\n")
0