結果
問題 |
No.391 CODING WAR
|
ユーザー |
|
提出日時 | 2021-07-02 22:54:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 245 ms / 2,000 ms |
コード長 | 861 bytes |
コンパイル時間 | 124 ms |
コンパイル使用メモリ | 82,436 KB |
実行使用メモリ | 112,440 KB |
最終ジャッジ日時 | 2024-06-29 12:47:45 |
合計ジャッジ時間 | 4,333 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 16 |
ソースコード
#!/usr/bin/env python3 # from typing import * import sys import io import math import collections import decimal import itertools import bisect import heapq def input(): return sys.stdin.readline()[:-1] # sys.setrecursionlimit(1000000) # _INPUT = """# paste here... # """ # sys.stdin = io.StringIO(_INPUT) MOD = 1000000007 def comb(n, r): r = min(r, n - r) return (fact[n] * factinv[r] * factinv[n-r]) % MOD fact = [1, 1] # fact[n] = (n! mod p) factinv = [1, 1] # factinv[n] = ((n!)^(-1) mod p) inv = [0, 1] # factinv 計算用 N, M = map(int, input().split()) for i in range(2, M + 1): fact.append((fact[-1] * i) % MOD) inv.append((-inv[MOD % i] * (MOD // i)) % MOD) factinv.append((factinv[-1] * inv[-1]) % MOD) ans = 0 for k in range(M): ans += (-1)**k * comb(M, k) * pow(M-k, N, MOD) ans %= MOD print(ans)