結果

問題 No.1318 ABCD quadruplets
ユーザー gew1fw
提出日時 2025-06-12 16:31:04
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 830 bytes
コンパイル時間 476 ms
コンパイル使用メモリ 82,280 KB
実行使用メモリ 84,536 KB
最終ジャッジ日時 2025-06-12 16:31:21
合計ジャッジ時間 5,362 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 11 TLE * 1 -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys

def main():
    N, M = map(int, sys.stdin.readline().split())
    ans = [0] * (N + 1)
    
    for d in range(M + 1):
        q_d = d * d
        if q_d > N:
            continue
        for c in range(M + 1):
            q_c = q_d + c * c + c * d
            if q_c > N:
                continue
            sum_cd = c + d
            for b in range(M + 1):
                sum_b = b * sum_cd + b * b
                q_b = q_c + sum_b
                if q_b > N:
                    continue
                sum_bcd = sum_cd + b
                for a in range(M + 1):
                    sum_a = a * a + a * sum_bcd
                    total = q_b + sum_a
                    if total <= N:
                        ans[total] += 1
    
    for count in ans:
        print(count)

if __name__ == "__main__":
    main()
0