Yukicoder No.462 6日知らずのコンピュータ

,

https://yukicoder.me/problems/no/462

悪い問題ではないと思います。

反省

入力は2行からなります。

と言っているのだから、そうでなくても常識的に、$k = 0$のとき$2$行目は存在しないのでなく空行が与えられるべきでは。$1$RE生えた。

$a_1, a_2, \dots, a_k$はこの順に使わなければいけないと誤読して$1$WA。これは私が悪い。

solution

やるだけ。

implementation

#!/usr/bin/env python3
mod = 10**9+7
def fact(n):
    if n == 0:
        return 1
    else:
        return n * fact(n-1) % mod
def bits(n):
    return set([ i for i, bit in enumerate(reversed(bin(n)[2 :])) if bit == '1' ])
n, k = map(int, input().split())
if k == 0:
    a = []
else:
    a = sorted(map(int, input().split()))
cnt = 1
for x, y in zip([ 0 ] + a, a + [ 2**n-1 ]):
    if bits(x).issubset(bits(y)):
        cnt = cnt * fact(len(bits(y)) - len(bits(x))) % mod
    else:
        cnt = 0
        break
print(cnt)