yukicoder No.133 カードゲーム

qiita.com

続きです。

yukicoder.me

問題概要

1 ~ 2 * N が書かれたカードを 2 分割して、N ターンあるごとに、1 枚ずつ出していく。大きい数が書かれていた方が勝ち。カードに書かれている番号が与えられるので、どちらかが勝つ確率を求める。

考えたこと

それぞれの手の出し方を求めていく。0 ~ n-1 の並べ替えをそれぞれ作って比較すれば良い。読みやすさを考えて、プレイヤーごとに再帰関数を分けた。

from math import factorial

used1 = []
used2 = []
n = 0
ap = []
bp = []
a, b = ([], [])

def rec2(pos):
    ans = 0
    if pos == n:
        # a, b の勝ち数
        aa, bb = (0, 0)
        for x, y in zip(ap, bp):
            if a[x] > b[y]:
                aa += 1
            elif a[x] < b[y]:
                bb += 1
        return 1 if aa > bb else 0
    for i in range(n):
        if not used2[i]:
            bp.append(i)
            used2[i] = True
            ans += rec2(pos+1)
            used2[i] = False
            bp.pop()
    return ans


def rec1(pos):
    ans = 0
    if pos == n:
        # return rec2(0, a, b)
        return rec2(0)
    for i in range(n):
        if not used1[i]:
            used1[i] = True
            ap.append(i)
            ans += rec1(pos+1)
            ap.pop()
            used1[i] = False
    return ans


def main():
    global used1, used2, n, a, b
    n = int(input())
    used1 = [False for i in range(n)]
    used2 = [False for i in range(n)]
    a = [int(i) for i in input().split()]
    b = [int(i) for i in input().split()]
    print(rec1(0)/factorial(n)**2)


if __name__ == '__main__':
    main()

22 ms で通った。今日の競プロは終了です。ありがとうございました。