結果

問題 No.1022 Power Equation
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-10-18 21:03:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,112 ms / 2,000 ms
コード長 1,429 bytes
コンパイル時間 606 ms
コンパイル使用メモリ 82,472 KB
実行使用メモリ 169,824 KB
最終ジャッジ日時 2024-10-18 21:03:21
合計ジャッジ時間 6,254 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
52,724 KB
testcase_01 AC 45 ms
61,460 KB
testcase_02 AC 74 ms
73,684 KB
testcase_03 AC 87 ms
76,300 KB
testcase_04 AC 807 ms
139,136 KB
testcase_05 AC 1,112 ms
169,824 KB
testcase_06 AC 1,094 ms
168,872 KB
testcase_07 AC 1,065 ms
167,116 KB
testcase_08 AC 566 ms
116,948 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/1022

import math


def calc_gcd(A, B):
    """
    正の整数A, Bの最大公約数を計算する
    """
    a = max(A, B)
    b = min(A, B)
    while a % b > 0:
        c = a % b
        a = b
        b = c
    return b

def solve(N):
    sqrt_n = int(math.sqrt(N))

    answer = (N ** 2)
    used_over_sqrt_n = set()
    used_under_sqrt_n = set()
    for a in range(2, sqrt_n + 1):
        if a in used_under_sqrt_n:
            continue

        arr = []
        x = a 
        while x <= N:
            arr.append(x)
            x *= a

        for x in arr:
            if x <= sqrt_n:
                used_under_sqrt_n.add(x)
            else:
                used_over_sqrt_n.add(x)
        
        x = len(arr)
        for m in range(1, x + 1):
            for n in range(1, x + 1):
                gcd = calc_gcd(m, n)

                lcm = (m // gcd) * n

                m1 = lcm // m
                n1 = lcm // n
                max_mn = max(m1, n1)
                ans = N // max_mn
                answer += ans
    
    n0 = N - sqrt_n - len(used_over_sqrt_n)
    return answer + n0 * N

        
            





def main():
    T = int(input())
    answers = []
    for _ in range(T):
        N = int(input())
        ans = solve(N)
        answers.append(ans)

    for ans in answers:
        print(ans)



        





    



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