結果

問題 No.1115 二つの数列 / Two Sequences
ユーザー hiraku
提出日時 2021-05-24 21:34:28
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 943 ms / 2,000 ms
コード長 1,367 bytes
コンパイル時間 135 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 105,852 KB
最終ジャッジ日時 2024-10-13 05:50:54
合計ジャッジ時間 16,011 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict, deque, Counter

# ２つの数列　隣接swap　一致させる　要素に重複なし


# 転倒数
def mergeCount(A):
    cnt = 0
    n = len(A)
    if n > 1:
        A1 = A[:n >> 1]
        A2 = A[n >> 1:]
        cnt += mergeCount(A1)
        cnt += mergeCount(A2)
        i1 = 0
        i2 = 0
        for i in range(n):
            if i2 == len(A2):
                A[i] = A1[i1]
                i1 += 1
            elif i1 == len(A1):
                A[i] = A2[i2]
                i2 += 1
            elif A1[i1] <= A2[i2]:
                A[i] = A1[i1]
                i1 += 1
            else:
                A[i] = A2[i2]
                i2 += 1
                cnt += n//2 - i1
    return cnt


n = int(input())
A = list(map(int, input().split()))
B = list(map(int, input().split()))

if Counter(A) != Counter(B):
    print(-1)
    exit()

# print(A)
# print(B)

# A[i] = B[s_i] となるような配列 S (s_0, s_1, ,,,)を考える
# A[i], B[i]で要素の値が等しいものが複数あるときは、i < j, s_i < s_jとなるように決める
# 決定したSに対して転倒数を求める

# Bの要素の値ごとの位置を調べる
pos = defaultdict(deque)
for itr, b in enumerate(B):
    pos[b].append(itr)

S = []
for a in A:
    p = pos[a].popleft()
    S.append(p)
# print(S)

print(mergeCount(S))
0