結果

問題 No.2449 square_permutation
ユーザー rlangevinrlangevin
提出日時 2023-09-09 00:55:28
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 889 bytes
コンパイル時間 540 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 320,268 KB
最終ジャッジ日時 2024-06-26 18:06:53
合計ジャッジ時間 10,853 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
54,400 KB
testcase_01 AC 39 ms
54,400 KB
testcase_02 AC 43 ms
54,272 KB
testcase_03 AC 41 ms
54,528 KB
testcase_04 AC 41 ms
54,656 KB
testcase_05 AC 54 ms
66,688 KB
testcase_06 AC 200 ms
155,712 KB
testcase_07 AC 206 ms
164,692 KB
testcase_08 AC 231 ms
174,004 KB
testcase_09 AC 250 ms
188,404 KB
testcase_10 AC 332 ms
234,648 KB
testcase_11 AC 361 ms
249,704 KB
testcase_12 AC 375 ms
260,624 KB
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 AC 39 ms
54,016 KB
testcase_23 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import *

def Sieve(n):
    lst = [True] * (n + 1)
    S = set()
    for i in range(2, ceil(sqrt(n)) + 1):
        if lst[i]:
            for j in range(2 * i, n + 1, i):
                lst[j] = False
    for i in range(2, n + 1):
        if lst[i]:
            S.add(i)
    return sorted(list(S))

N = int(input())
S = []
for i in Sieve(500):
    S.append(i * i)
    
L = []
for i in range(N + 1):
    L.append([i, i])
    
for s in S:
    now = s
    while now <= N + 1:
        for i in range(now, N + 1, now):
            L[i][0] //= s
        now *= s
        
from collections import *
ans = list(range(N + 1))
D = [deque() for i in range(N + 1)]
for i in range(N + 1):
    D[L[i][0]].append(L[i][1])
    
for i in range(1, N + 1):
    while len(D[i]) >= 2:
        a = D[i].popleft()
        b = D[i].pop()
        ans[a], ans[b] = ans[b], ans[a]
        
print(*ans[1:])
0