結果
問題 | No.140 みんなで旅行 |
ユーザー |
|
提出日時 | 2020-11-21 00:10:27 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 464 ms / 5,000 ms |
コード長 | 1,246 bytes |
コンパイル時間 | 481 ms |
コンパイル使用メモリ | 82,648 KB |
実行使用メモリ | 119,936 KB |
最終ジャッジ日時 | 2024-07-23 14:09:19 |
合計ジャッジ時間 | 4,723 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 19 |
ソースコード
import sysinput = lambda : sys.stdin.readline().rstrip()sys.setrecursionlimit(max(1000, 10**9))write = lambda x: sys.stdout.write(x+"\n")n = int(input())i = 0dp = [[0]*(2*i+1) for _ in range(i+1)]dp[0][0] = 1M = 10**9+7### 素数の逆元とCombinationN = 2*n+10 # 必要なテーブルサイズg1 = [None] * (N+1) # 元テーブルg2 = [None] * (N+1) #逆元テーブルinverse = [None] * (N+1) #逆元テーブル計算用テーブルg1[0] = g1[1] = g2[0] = g2[1] = 1inverse[0], inverse[1] = [0, 1]for i in range( 2, N + 1 ):g1[i] = ( g1[i-1] * i ) % Minverse[i] = ( -inverse[M % i] * (M//i) ) % M # ai+b==0 mod M <=> i==-b*a^(-1) <=> i^(-1)==-b^(-1)*aよりg2[i] = (g2[i-1] * inverse[i]) % Mdef cmb(n, r, M):if ( r<0 or r>n ):return 0r = min(r, n-r)return (g1[n] * g2[r] * g2[n-r]) % Mdef perm(n, r, M):if (r<0 or r>n):return 0return (g1[n] * g2[n-r]) % Mans = 0s = [[0]*(n+1) for _ in range(n+1)]s[0][0] = 1for i in range(1,n+1):for j in range(1,n+1):s[i][j] = s[i-1][j-1] + j*s[i-1][j]for k in range(1, n+1):for i in range(k,n+1):ans += s[i][k]*pow(perm(k,2,M), n-i, M)*cmb(n,i,M)ans %= M# print(ans)print(ans%M)