結果

問題 No.1318 ABCD quadruplets
ユーザー lam6er
提出日時 2025-04-15 23:35:18
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 977 bytes
コンパイル時間 181 ms
コンパイル使用メモリ 81,868 KB
実行使用メモリ 84,104 KB
最終ジャッジ日時 2025-04-15 23:36:17
合計ジャッジ時間 4,635 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
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):
        s1 = d * d
        if s1 > N:
            continue
        for c in range(M + 1):
            sum_cd = c + d
            term_c = c * sum_cd
            s2 = s1 + term_c
            if s2 > N:
                break
            for b in range(M + 1):
                sum_bcd = b + c + d
                term_b = b * sum_bcd
                s3 = s2 + term_b
                if s3 > N:
                    break
                # Precompute sum_bcd for a's loop
                sum_bcd_val = sum_bcd
                for a in range(M + 1):
                    a_plus_sum = a + sum_bcd_val
                    term_a = a * a_plus_sum
                    s4 = s3 + term_a
                    if s4 > N:
                        break
                    ans[s4] += 1
    
    print('\n'.join(map(str, ans)))

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