結果

問題 No.1318 ABCD quadruplets
コンテスト
ユーザー ああいい
提出日時 2022-01-29 19:50:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,675 ms / 2,000 ms
コード長 1,403 bytes
コンパイル時間 981 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 78,592 KB
最終ジャッジ日時 2025-01-02 14:58:15
合計ジャッジ時間 16,140 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,M = map(int,input().split())

def f(a,b,c,d):
    return a * (a + b + c + d) + b * (b + c + d) + c * (c + d) + d * d
def count(a,b,c,d):
    num = 4
    if a == b:num -= 1
    if b == c:num -= 1
    if c == d:num -= 1
    if num == 1:return 1
    if num == 3:return 12
    if num == 4:return 24
    if a < b or c < d:return 4
    return 6
dat = [0] * (N + 1)
from collections import defaultdict
for a in range(M+1):
    for b in range(a,M+1):
        if f(a,b,b,b) > N:break
        for c in range(b,M+1):
            if f(a,b,c,c) > N:break
            for d in range(c,M+1):
                u = f(a,b,c,d)
                if u <= N:
                    """
                    s = defaultdict(int)
                    s[a] += 1
                    s[b] += 1
                    s[c] += 1
                    s[d] += 1
                    e = 0
                    if len(s) == 1:
                        e = 1
                    elif len(s) == 3:
                        e = 12
                    elif len(s) == 4:
                        e = 24
                    else:
                        if s[a] == 1 or s[a] == 3:
                            e = 4
                        else:
                            e = 6
                    dat[u] += e
                    """
                    dat[u] += count(a,b,c,d)
                else:
                    break
for i in dat:
    print(i)
0