結果

問題 No.2979 直角三角形の個数
ユーザー lif4635lif4635
提出日時 2024-12-03 02:47:07
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 3,938 bytes
コンパイル時間 409 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 85,000 KB
最終ジャッジ日時 2024-12-03 02:47:35
合計ジャッジ時間 26,145 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
52,352 KB
testcase_01 AC 36 ms
52,608 KB
testcase_02 AC 37 ms
52,736 KB
testcase_03 AC 41 ms
52,352 KB
testcase_04 AC 40 ms
52,352 KB
testcase_05 AC 39 ms
53,020 KB
testcase_06 AC 39 ms
53,376 KB
testcase_07 AC 38 ms
52,992 KB
testcase_08 AC 51 ms
61,568 KB
testcase_09 AC 45 ms
60,544 KB
testcase_10 AC 70 ms
71,552 KB
testcase_11 AC 53 ms
63,616 KB
testcase_12 AC 114 ms
76,416 KB
testcase_13 AC 86 ms
76,160 KB
testcase_14 AC 292 ms
76,584 KB
testcase_15 AC 284 ms
76,544 KB
testcase_16 AC 1,255 ms
76,800 KB
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 AC 40 ms
52,736 KB
testcase_25 AC 38 ms
52,736 KB
testcase_26 AC 38 ms
53,248 KB
testcase_27 AC 132 ms
76,736 KB
testcase_28 WA -
権限があれば一括ダウンロードができます

ソースコード

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

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)
    if rq - lq <= const:
        for r in range(lq,rq,2):
            if not (0 < r - p < p):
                continue
            for i in fp:
                if r%i == 0:
                    break
            else:
                ans += m//r #この倍数まで大丈夫
    else:
        #この時あきらかに const <= lq
        olq,orq = lq,rq
        for res in range(max(m//rq,1), m//lq+1):
            rq = m // res
            lq = m // (res + 1) + 1
            #加算されるのがlの範囲
            lq = max(olq-1,lq)
            rq = min(rq,orq-1)
            if lq >= lq:
                continue
            
            #この内倍数でもなければ2の倍数でもないもの
            fp_ = [2] + fp
            while fp_ and fp_[-1] > rq:
                fp_.pop()
            
            cnt = rq - lq
            l_ = len(fp)
            for i in range(l_):
                r = fp_[i]
                if rq < r:
                    break
                cnt -= rq//r - lq//r
            for i in range(l_):
                r = fp_[i]
                if rq < r:
                    break
                for j in range(i+1,l_):
                    s = fp_[j]
                    if rq < r*s:
                        break
                    cnt += rq//(r*s) - lq//(r*s)
            for i in range(l_):
                r = fp_[i]
                if rq < r:
                    break
                for j in range(i+1,l_):
                    s = fp_[j]
                    if rq < r*s:
                        break
                    for k in range(j+1,l_):
                        t = fp_[k]
                        if rq < r*s*t:
                            break
                        cnt -= rq//(r*s*t) - lq//(r*s*t)
            for i in range(l_):
                r = fp_[i]
                if rq < r:
                    break
                for j in range(i+1,l_):
                    s = fp_[j]
                    if rq < r*s:
                        break
                    for k in range(j+1,l_):
                        t = fp_[k]
                        if rq < r*s*t:
                            break
                        for g in range(k+1,l_):
                            u = fp_[g]
                            if rq < r*s*t*u:
                                break
                            cnt += rq//(r*s*t*u) - lq//(r*s*t*u)
            for i in range(l_):
                r = fp_[i]
                if rq < r:
                    break
                for j in range(i+1,l_):
                    s = fp_[j]
                    if rq < r*s:
                        break
                    for k in range(j+1,l_):
                        t = fp_[k]
                        if rq < r*s*t:
                            break
                        for g in range(k+1,l_):
                            u = fp_[g]
                            if rq < r*s*t*u:
                                break
                            for h in range(g+1,l_):
                                v = fp_[h]
                                if rq < r*s*t*u*v:
                                    break
                                cnt -= rq//(r*s*t*u*v) - lq//(r*s*t*u*v)
            ans += cnt*res                
        
print(ans) 
0