結果

問題 No.767 配られたジャパリまん
ユーザー nephrologistnephrologist
提出日時 2022-03-23 23:24:42
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,977 bytes
コンパイル時間 204 ms
コンパイル使用メモリ 82,376 KB
実行使用メモリ 137,912 KB
最終ジャッジ日時 2024-10-12 02:51:41
合計ジャッジ時間 4,755 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
53,004 KB
testcase_01 AC 36 ms
52,712 KB
testcase_02 AC 37 ms
54,488 KB
testcase_03 AC 37 ms
53,008 KB
testcase_04 AC 71 ms
72,736 KB
testcase_05 AC 277 ms
82,692 KB
testcase_06 AC 36 ms
52,684 KB
testcase_07 AC 37 ms
53,872 KB
testcase_08 AC 41 ms
58,788 KB
testcase_09 AC 70 ms
72,988 KB
testcase_10 AC 127 ms
79,860 KB
testcase_11 AC 78 ms
76,704 KB
testcase_12 AC 75 ms
75,844 KB
testcase_13 AC 126 ms
79,516 KB
testcase_14 AC 74 ms
73,864 KB
testcase_15 AC 47 ms
63,044 KB
testcase_16 AC 76 ms
76,136 KB
testcase_17 AC 43 ms
60,980 KB
testcase_18 AC 44 ms
60,796 KB
testcase_19 AC 84 ms
78,724 KB
testcase_20 TLE -
権限があれば一括ダウンロードができます

ソースコード

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


ans = [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
    ans[i] = temp

total = comb(h + w, w)


# ans = ans[:]

# 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 * ans[j]
#         ans[i] %= mod
#         # print(keisu * ans[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]
print(*ans, sep="\n")
0