結果

問題 No.1779 Magical Swap
ユーザー WizistWizist
提出日時 2021-12-12 22:44:40
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 469 ms / 2,000 ms
コード長 878 bytes
コンパイル時間 494 ms
コンパイル使用メモリ 82,316 KB
実行使用メモリ 131,556 KB
最終ジャッジ日時 2024-07-21 11:11:04
合計ジャッジ時間 4,137 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#!/usr/bin/env python3
#
# No.1779 Magical Swap
#
import math
N = 100001
fp = [0] * N
for i in range(2, len(fp)):
if fp[i] == 0:
fp[i] = i
for j in range(i * i, len(fp), i):
if fp[j] == 0: fp[j] = i
def lcm(x, y): return x * y // math.gcd(x, y)
ncases = int(input())
for caseno in range(ncases):
n = int(input())
a = list(map(int, input().split()))
b = list(map(int, input().split()))
c1, c2 = {}, {}
for i in range(1, len(a)):
if a[i] == b[i]: continue
if a[i] not in c1: c1[a[i]] = []
if b[i] not in c2: c2[b[i]] = []
c1[a[i]].append(fp[i + 1])
c2[b[i]].append(fp[i + 1])
def check():
if a[0] != b[0]: return False
for x in c1:
a1 = c1.get(x, []); a2 = c2.get(x, [])
if len(a1) != len(a2): return False
a1.sort(); a2.sort()
if lcm(a1[-1], 2) > n or lcm(a2[-1], 2) > n: return False
return True
print("Yes" if check() else "No")
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0