結果
問題 | No.1035 Color Box |
ユーザー |
|
提出日時 | 2021-07-02 23:03:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 230 ms / 2,000 ms |
コード長 | 861 bytes |
コンパイル時間 | 163 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 112,108 KB |
最終ジャッジ日時 | 2024-06-29 12:55:19 |
合計ジャッジ時間 | 7,049 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
#!/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)