結果

問題 No.1318 ABCD quadruplets
コンテスト
ユーザー convexineq
提出日時 2020-12-15 22:30:22
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,774 bytes
コンパイル時間 137 ms
コンパイル使用メモリ 82,316 KB
実行使用メモリ 83,476 KB
最終ジャッジ日時 2024-09-20 02:23:10
合計ジャッジ時間 8,314 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20 TLE * 1 -- * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,m = map(int,input().split())
ans = [0]*(n+1)
# all different
for a in range(m+1):
    v0 = a*a
    for b in range(a+1,m+1):
        v1 = v0 + b*(a+b)
        if v1 > n: break
        for c in range(b+1,m+1):
            v2 = v1 + c*(a+b+c)
            if v2 > n: break
            for d in range(c+1,m+1):
                v3 = v2 + d*(a+b+c+d)
                if v3 <= n:
                    ans[v3] += 24
# a=b<c<d
for a in range(m+1):
    v1 = 3*a*a
    if v1 > n: break
    for c in range(a+1,m+1):
        v2 = v1 + c*(a+a+c)
        if v2 > n: break
        for d in range(c+1,m+1):
            v3 = v2 + d*(a+a+c+d)
            if v3 <= n:
                ans[v3] += 12
# a<b=c<d
for a in range(m+1):
    for b in range(a+1,m+1):
        v2 = a*a + b*(2*a+3*b)
        if v2 > n: break
        for d in range(b+1,m+1):
            v3 = v2 + d*(a+b+b+d)
            if v3 <= n:
                ans[v3] += 12
# a<b<c=d
for a in range(m+1):
    v0 = a*a
    for b in range(a+1,m+1):
        v1 = v0 + b*(a+b)
        if v1 > n: break
        for c in range(b+1,m+1):
            v3 = v1 + c*(2*a+2*b+3*c)
            if v3 <= n:
                ans[v3] += 12
# a=b=c<d
for a in range(m+1):
    v2 = 6*a*a
    if v2 > n: break
    for d in range(a+1,m+1):
        v3 = v2 + d*(3*a+d)
        if v3 <= n:
            ans[v3] += 4
# a<b=c=d
for d in range(m+1):
    v2 = 6*d*d
    if v2 > n: break
    for a in range(d):
        v3 = v2 + a*(3*d+a)
        if v3 <= n:
            ans[v3] += 4
# a=b<c=d
for a in range(m+1):
    v1 = 3*a*a
    if v1 > n: break
    for c in range(a+1,m+1):
        v3 = v1 + 4*a*c + 3*c*c
        if v3 <= n:
            ans[v3] += 6
# a=b=c=d
for a in range(m+1):
    v3 = 10*a*a
    if v3 <= n:
        ans[v3] += 1

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