結果
問題 |
No.1022 Power Equation
|
ユーザー |
|
提出日時 | 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 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 8 |
ソースコード
## 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()