結果

問題 No.1115 二つの数列 / Two Sequences
ユーザー ryo_nryo_n
提出日時 2020-07-18 01:57:28
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 1,080 ms / 2,000 ms
コード長 1,508 bytes
コンパイル時間 82 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 30,548 KB
最終ジャッジ日時 2024-05-07 15:00:58
合計ジャッジ時間 17,550 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 86 ms
12,288 KB
testcase_01 AC 27 ms
10,880 KB
testcase_02 AC 27 ms
10,880 KB
testcase_03 AC 895 ms
27,076 KB
testcase_04 AC 1,043 ms
30,524 KB
testcase_05 AC 935 ms
27,000 KB
testcase_06 AC 833 ms
26,484 KB
testcase_07 AC 1,080 ms
30,548 KB
testcase_08 AC 70 ms
11,776 KB
testcase_09 AC 577 ms
22,432 KB
testcase_10 AC 999 ms
30,524 KB
testcase_11 AC 27 ms
10,624 KB
testcase_12 AC 1,045 ms
30,524 KB
testcase_13 AC 1,023 ms
30,524 KB
testcase_14 AC 1,006 ms
30,524 KB
testcase_15 AC 28 ms
10,880 KB
testcase_16 AC 28 ms
10,752 KB
testcase_17 AC 28 ms
10,880 KB
testcase_18 AC 28 ms
10,880 KB
testcase_19 AC 28 ms
10,880 KB
testcase_20 AC 27 ms
10,880 KB
testcase_21 AC 26 ms
10,624 KB
testcase_22 AC 28 ms
10,752 KB
testcase_23 AC 91 ms
12,672 KB
testcase_24 AC 362 ms
18,584 KB
testcase_25 AC 769 ms
24,260 KB
testcase_26 AC 173 ms
14,532 KB
testcase_27 AC 437 ms
21,232 KB
testcase_28 AC 598 ms
22,872 KB
testcase_29 AC 814 ms
26,504 KB
testcase_30 AC 975 ms
30,432 KB
testcase_31 AC 239 ms
15,704 KB
testcase_32 AC 160 ms
14,188 KB
testcase_33 AC 831 ms
27,268 KB
testcase_34 AC 28 ms
10,880 KB
testcase_35 AC 26 ms
10,880 KB
testcase_36 AC 26 ms
10,752 KB
testcase_37 AC 26 ms
10,752 KB
testcase_38 AC 26 ms
10,752 KB
testcase_39 AC 26 ms
10,880 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

sys.setrecursionlimit(10 ** 8)

input = sys.stdin.readline


def main():
    N = int(input())
    A = [int(x) for x in input().split()]
    B = [int(x) for x in input().split()]

    dic = {}
    for i, b in enumerate(B):
        dic[b] = i

    C = []
    for a in A:
        C.append(dic[a])



    def segfunc(x,y):
       return x + y

    def init(init_val):
        for i in range(n):
            seg[i + num - 1] = init_val[i]

        for i in range(num - 2, -1, -1):
            seg[i] = segfunc(seg[2 * i + 1], seg[2 * i + 2])


    def update(k, x):
        k += num - 1
        seg[k] = x
        while k:
            k = (k - 1) // 2
            seg[k] = segfunc(seg[k * 2 + 1], seg[k * 2 + 2])


    def query(p, q):
        if q <= p:
            return ide_ele
        p += num - 1
        q += num - 2
        res = ide_ele
        while q - p > 1:
            if p & 1 == 0:
                res = segfunc(res, seg[p])
            if q & 1 == 1:
                res = segfunc(res, seg[q])
                q -= 1
            p = p // 2
            q = (q - 1) // 2
        if p == q:
            res = segfunc(res, seg[p])
        else:
            res = segfunc(segfunc(res, seg[p]), seg[q])
        return res


    # seg tree初期値 (単位元)
    n = len(C)
    ide_ele = 0
    num = 2 ** (n - 1).bit_length()
    seg = [ide_ele] * 2 * num

    ans = 0
    for c in C:
        ans += query(c, N)
        update(c, 1)

    print(ans)


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