結果

問題 No.1035 Color Box
ユーザー hedwig100
提出日時 2020-04-24 22:30:24
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 378 ms / 2,000 ms
コード長 969 bytes
コンパイル時間 148 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 18,816 KB
最終ジャッジ日時 2024-10-15 03:11:55
合計ジャッジ時間 7,616 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

INF = 10 ** 9
MOD = 10 **9 + 7
import sys
sys.setrecursionlimit(100000000)
dy = (-1,0,1,0)
dx = (0,1,0,-1)
from  collections import deque
from math import factorial

MAXN = 100005
factorial = [1]
for i in range(1, MAXN + 1):
    factorial.append(factorial[-1] * i % MOD)

inv_factorial = [-1] * (MAXN + 1)
inv_factorial[-1] = pow(factorial[-1], MOD-2, MOD)
for i in reversed(range(MAXN)):
    inv_factorial[i] = inv_factorial[i + 1] * (i + 1) % MOD

def fact(n):
    return factorial[n]

def nck(n, k):
    if k>n or k<0:
        return 0
    else:
        return factorial[n] * inv_factorial[n-k] * inv_factorial[k]%MOD

def main():
    n,m = map(int,input().split())
    ans = 0
    for i in range(m):
        if i%2 == 0:
            ans += nck(m,i) * pow(m - i,n,MOD) % MOD
            ans %= MOD
        else:
            ans -= nck(m,i) * pow(m - i,n,MOD) % MOD 
            ans += MOD
            ans %= MOD   
    print(ans)
if __name__=='__main__':
    main()
0