結果

問題 No.2979 直角三角形の個数
ユーザー lif4635lif4635
提出日時 2024-12-03 23:17:13
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 3,380 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 82,412 KB
実行使用メモリ 141,908 KB
最終ジャッジ日時 2024-12-03 23:17:49
合計ジャッジ時間 35,842 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
59,928 KB
testcase_01 AC 37 ms
60,368 KB
testcase_02 AC 37 ms
60,716 KB
testcase_03 AC 40 ms
135,652 KB
testcase_04 AC 39 ms
60,560 KB
testcase_05 AC 39 ms
139,828 KB
testcase_06 AC 43 ms
63,748 KB
testcase_07 AC 41 ms
141,908 KB
testcase_08 AC 101 ms
83,180 KB
testcase_09 AC 67 ms
72,016 KB
testcase_10 AC 153 ms
77,184 KB
testcase_11 AC 127 ms
77,152 KB
testcase_12 AC 201 ms
77,996 KB
testcase_13 AC 170 ms
77,800 KB
testcase_14 AC 293 ms
78,732 KB
testcase_15 AC 291 ms
78,264 KB
testcase_16 AC 514 ms
78,096 KB
testcase_17 AC 955 ms
78,664 KB
testcase_18 AC 2,478 ms
79,668 KB
testcase_19 WA -
testcase_20 TLE -
testcase_21 TLE -
testcase_22 TLE -
testcase_23 TLE -
testcase_24 AC 38 ms
53,876 KB
testcase_25 AC 36 ms
53,392 KB
testcase_26 AC 38 ms
53,352 KB
testcase_27 AC 238 ms
77,660 KB
testcase_28 TLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import isqrt

n = int(input())

rootn = isqrt(n)
minp = [0]*(rootn+1)

# 素因数の前計算
for i in range(3,rootn+1,2):
    if minp[i] != 0:
        continue
    for j in range(i,rootn+1,i):
        if minp[j] == 0:
            minp[j] = i

from itertools import permutations

def calc(x,fp):
    cnt = x
    
    l = len(fp)
    for i in range(l):
        r = fp[i]
        if x < r:
            break
        cnt -= x//r
    for i in range(l):
        r = fp[i]
        if x < r:
            break
        for j in range(i+1,l):
            s = fp[j]
            if x < r*s:
                break
            cnt += x//(r*s)
    for i in range(l):
        r = fp[i]
        if x < r:
            break
        for j in range(i+1,l):
            s = fp[j]
            if x < r*s:
                break
            for k in range(j+1,l):
                t = fp[k]
                if x < r*s*t:
                    break
                cnt -= x//(r*s*t)
    for i in range(l):
        r = fp[i]
        if x < r:
            break
        for j in range(i+1,l):
            s = fp[j]
            if x< r*s:
                break
            for k in range(j+1,l):
                t = fp[k]
                if x < r*s*t:
                    break
                for g in range(k+1,l):
                    u = fp[g]
                    if x < r*s*t*u:
                        break
                    cnt += x//(r*s*t*u)
    for i in range(l):
        r = fp[i]
        if x < r:
            break
        for j in range(i+1,l):
            s = fp[j]
            if x < r*s:
                break
            for k in range(j+1,l):
                t = fp[k]
                if x < r*s*t:
                    break
                for g in range(k+1,l):
                    u = fp[g]
                    if x < r*s*t*u:
                        break
                    for h in range(g+1,l):
                        v = fp[h]
                        if x < r*s*t*u*v:
                            break
                        cnt -= x//(r*s*t*u*v)
    return cnt

ans = 0
#pの全探索
const = 10**4
for p in range(2,rootn + 1):
    fp = []
    p_ = p
    while minp[p_] != 0:
        mp = minp[p_]
        fp.append(mp)
        while p_%mp == 0:
            p_ //= mp
    
    # q は奇数
    # p < q < 2p
    # maxの値はm以下
    m = n//(2*p)
    
    lim = isqrt(m) #ここを境界とします
    lq = 2 * ((p+1)//2) + 1
    rq = min(m+1,2*p)
    
    for r in range(lq,rq,2):
        if m//r <= lim:
            break
        
        if not (0 < r - p < p):
            continue
        for i in fp:
            if r%i == 0:
                break
        else:
            ans += m//r #この倍数まで大丈夫
    
    #この時あきらかに const <= lq
    olq,orq = lq,rq
    for res in range(max(m//orq,1), lim+1):
        lq = m // (res + 1)
        rq = m // res
        #加算されるのがlの範囲
        lq = max(lq,olq-1)
        rq = min(rq,orq-1)
        if lq >= rq:
            continue
        
        # print(res,p,lq,rq)
        #この内倍数でもなければ2の倍数でもないもの
        fp_ = [2] + fp
        while fp_ and fp_[-1] > rq:
            fp_.pop()
        
        assert fp_.count(2) <= 1
        cnt = calc(rq,fp_[:]) - calc(lq,fp_[:])
        ans += cnt*res                
        
print(ans)
0