結果

問題 No.140 みんなで旅行
ユーザー titiatitia
提出日時 2022-03-08 05:12:12
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 779 ms / 5,000 ms
コード長 828 bytes
コンパイル時間 146 ms
コンパイル使用メモリ 11,004 KB
実行使用メモリ 73,388 KB
最終ジャッジ日時 2023-09-30 09:14:17
合計ジャッジ時間 8,154 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 136 ms
24,296 KB
testcase_01 AC 136 ms
24,376 KB
testcase_02 AC 163 ms
25,704 KB
testcase_03 AC 136 ms
24,340 KB
testcase_04 AC 139 ms
24,204 KB
testcase_05 AC 138 ms
24,324 KB
testcase_06 AC 139 ms
24,256 KB
testcase_07 AC 140 ms
24,380 KB
testcase_08 AC 138 ms
24,336 KB
testcase_09 AC 136 ms
24,380 KB
testcase_10 AC 135 ms
24,376 KB
testcase_11 AC 779 ms
73,304 KB
testcase_12 AC 156 ms
25,280 KB
testcase_13 AC 142 ms
24,592 KB
testcase_14 AC 774 ms
73,388 KB
testcase_15 AC 768 ms
73,084 KB
testcase_16 AC 343 ms
38,696 KB
testcase_17 AC 225 ms
30,116 KB
testcase_18 AC 615 ms
59,952 KB
testcase_19 AC 674 ms
64,752 KB
testcase_20 AC 211 ms
29,492 KB
testcase_21 AC 136 ms
24,344 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 解説を見た。
# 包除っぽく見えたが間違いだった。

N=int(input())
mod=10**9+7

FACT=[1]
for i in range(1,2*10**5+1):
    FACT.append(FACT[-1]*i%mod)

FACT_INV=[pow(FACT[-1],mod-2,mod)]
for i in range(2*10**5,0,-1):
    FACT_INV.append(FACT_INV[-1]*i%mod)

FACT_INV.reverse()

def Combi(a,b):
    if 0<=b<=a:
        return FACT[a]*FACT_INV[b]%mod*FACT_INV[a-b]%mod
    else:
        return 0

# 夫婦同じグループに入る夫婦がx組でグループ数がyとなる組み合わせ
from functools import lru_cache
@lru_cache(maxsize=None)
def F(x,y):
    if x==0 and y==0:
        return 1
    if x<y or y<=0:
        return 0
    return F(x-1,y-1)+y*F(x-1,y)

ANS=0

for x in range(N+1):
    for y in range(x+1):
        ANS+=F(x,y)*Combi(N,x)*pow(y*(y-1),N-x,mod)
        ANS%=mod

print(ANS)
0