結果

問題 No.767 配られたジャパリまん
ユーザー nephrologistnephrologist
提出日時 2022-03-23 23:22:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,984 ms / 2,000 ms
コード長 1,993 bytes
コンパイル時間 263 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 104,516 KB
最終ジャッジ日時 2024-04-20 07:37:53
合計ジャッジ時間 5,522 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
52,352 KB
testcase_01 AC 40 ms
51,968 KB
testcase_02 AC 37 ms
52,864 KB
testcase_03 AC 37 ms
52,736 KB
testcase_04 AC 73 ms
72,832 KB
testcase_05 AC 285 ms
81,036 KB
testcase_06 AC 36 ms
52,096 KB
testcase_07 AC 38 ms
52,992 KB
testcase_08 AC 41 ms
58,752 KB
testcase_09 AC 72 ms
72,832 KB
testcase_10 AC 130 ms
78,720 KB
testcase_11 AC 84 ms
78,464 KB
testcase_12 AC 79 ms
76,160 KB
testcase_13 AC 126 ms
78,848 KB
testcase_14 AC 76 ms
73,600 KB
testcase_15 AC 50 ms
61,696 KB
testcase_16 AC 80 ms
76,768 KB
testcase_17 AC 44 ms
59,904 KB
testcase_18 AC 46 ms
60,544 KB
testcase_19 AC 88 ms
78,976 KB
testcase_20 AC 1,984 ms
104,516 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 問題の首座として、通れない、としたときにどうなるか、を高速で求めたい。
# 通れないは難しいから、通るものを数えて、全体から引く
# 全体から引くときに、各々引いて、,,,とやるとかぶるから包除原理
import sys

input = sys.stdin.buffer.readline
mod = 10 ** 8 + 7
h, w, k = map(int, input().split())

AB = [list(map(int, input().split())) for i in range(k)]

# AB.sort(key=lambda x: [x[1], x[2]])
# idx = [i for i, _, __ in AB]

N = h + w + 100 + 5
bikkuri = [0] * N
bikkuri[0] = 1
gyaku = [0] * N
for i in range(1, N):
    bikkuri[i] = (i * bikkuri[i - 1]) % mod
gyaku[N - 1] = pow(bikkuri[N - 1], mod - 2, mod)
for i in range(N)[::-1]:
    gyaku[i - 1] = (gyaku[i] * i) % mod


def comb(n, r):
    if r < 0 or n - r < 0:
        return 0
    return bikkuri[n] * gyaku[r] % mod * gyaku[n - r] % mod


popcount = [0] * (1 << k)

for i in range(1 << k):
    popcount[i] = popcount[i // 2] + i % 2


memo = [0] * (1 << k)

for i in range(1 << k):
    temp = 1
    kouho = []
    for j in range(k):
        if (i >> j) & 1:
            kouho.append(AB[j])
    kouho.sort()
    px, py = 0, 0

    for nx, ny in kouho:
        dx = nx - px
        dy = ny - py
        temp *= comb(dx + dy, dy)
        temp %= mod
        px, py = nx, ny
    nx, ny = h, w

    dx = nx - px
    dy = ny - py
    keisu = 1 if popcount[i] % 2 else -1
    temp *= keisu * comb(dx + dy, dy)
    temp %= mod
    memo[i] = temp

total = comb(h + w, w)


ans = memo[:]

# for i in range(1 << k):
#     j = i
#     while j:
#         # print("i,j", i, j)
#         keisu = 1 if popcount[j] % 2 else -1
#         ans[i] += keisu * memo[j]
#         ans[i] %= mod
#         # print(keisu * memo[j])
#         j -= 1
#         j &= i

for i in range(k):
    for j in range(1 << k):
        if (j >> i) & 1:
            ans[j] += ans[j ^ (1 << i)]
            ans[j] %= mod


# ans = [total - a for a in ans]

for a in ans:
    print(-a % mod)
0