結果

問題 No.2130 分配方法の数え上げ mod 998244353
コンテスト
ユーザー scrappy
提出日時 2022-11-25 23:26:21
言語 Go
(1.23.4)
結果
WA  
実行時間 -
コード長 662 bytes
コンパイル時間 12,082 ms
コンパイル使用メモリ 223,408 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-02 05:56:27
合計ジャッジ時間 13,207 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 30 WA * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

// No.2130 分配方法の数え上げ mod 998244353
package main

import (
	"fmt"
)

const MOD = 998_244_353

func main() {
	var N, M int
	fmt.Scan(&N, &M)

	a := modpow(2, N, MOD)

	b := 1
	for i := 0; i < M; i++ {
		a = (a - b + MOD) % MOD

		b = (b * (N - i) % MOD) % MOD
		b = (b * modinv(i+1, MOD)) % MOD
	}
	fmt.Println(a)
}

func modinv(a, m int) int {
	b := m
	u := 1
	v := 0
	for b != 0 {
		t := a / b
		a -= t * b
		a, b = b, a
		u -= t * v
		u, v = v, u
	}
	u %= m
	if u < 0 {
		u += m
	}
	return u
}

func modpow(a int, n int, mod int) int {
	m := 1
	for n > 0 {
		if n&1 == 1 {
			m = (m * a) % mod
		}
		a = (a * a) % mod
		n >>= 1
	}
	return m
}
0