May 23, 2021

ABC202_C

Table of Contents

AtCoder Beginner Contest 202 C

お題

問題

ポイント

まずは、B, C から$$ B_{C_j} $$ の新しい数列を作ってしまい、どの数字がいくつあるかを数える。

あとはAを頭から順番に、何個あるか数える。

回答

from collections import Counter
def main():
    N = int(input())
    A = map(int, input().split())
    B = list(map(int, input().split()))
    C = map(int, input().split())

    BB = []
    for c in C:
        c = c - 1
        BB.append(B[c])

    count = Counter(BB)

    ans = 0
    for a in A:
        ans += count[a]

    print(ans)
if __name__ == "__main__":
    main()