結果

問題 No.2409 Strange Werewolves
ユーザー rulerruler
提出日時 2023-08-12 06:10:49
言語 Go
(1.22.1)
結果
AC  
実行時間 52 ms / 2,000 ms
コード長 2,758 bytes
コンパイル時間 15,184 ms
コンパイル使用メモリ 235,776 KB
実行使用メモリ 26,496 KB
最終ジャッジ日時 2024-11-19 03:56:58
合計ジャッジ時間 16,764 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 52 ms
26,496 KB
testcase_01 AC 51 ms
26,496 KB
testcase_02 AC 52 ms
26,496 KB
testcase_03 AC 51 ms
26,496 KB
testcase_04 AC 50 ms
26,496 KB
testcase_05 AC 51 ms
26,496 KB
testcase_06 AC 50 ms
26,496 KB
testcase_07 AC 52 ms
26,496 KB
testcase_08 AC 51 ms
26,496 KB
testcase_09 AC 52 ms
26,496 KB
testcase_10 AC 52 ms
26,496 KB
testcase_11 AC 52 ms
26,496 KB
testcase_12 AC 50 ms
26,496 KB
testcase_13 AC 51 ms
26,496 KB
testcase_14 AC 50 ms
26,496 KB
testcase_15 AC 52 ms
26,496 KB
testcase_16 AC 51 ms
26,496 KB
testcase_17 AC 52 ms
26,496 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

package main

import (
	"bufio"
	"fmt"
	"os"
	"sort"
	"strconv"
)

func Assert(ok bool) {
	if !ok {
		panic("assertion failed")
	}
}

var sc = func() *bufio.Scanner {
	sc := bufio.NewScanner(os.Stdin)
	sc.Split(bufio.ScanWords)
	return sc
}()

func Str() string {
	sc.Scan()
	return sc.Text()
}

func Int() int {
	x, _ := strconv.Atoi(Str())
	return x
}

func Float() float64 {
	x, _ := strconv.ParseFloat(Str(), 64)
	return x
}

var wr = bufio.NewWriter(os.Stdout)

func Out(a ...interface{}) {
	fmt.Fprintln(wr, a...)
}

func Format(a []int) string {
	s := fmt.Sprint(a)
	return s[1 : len(s)-1]
}

func Chmin(a *int, b int) bool {
	if b < *a {
		*a = b
		return true
	}
	return false
}

func Chmax(a *int, b int) bool {
	if b > *a {
		*a = b
		return true
	}
	return false
}

func Max(a ...int) int {
	mx := a[0]
	for _, x := range a {
		if x > mx {
			mx = x
		}
	}
	return mx
}

func Min(a ...int) int {
	mn := a[0]
	for _, x := range a {
		if x < mn {
			mn = x
		}
	}
	return mn
}

func Sum(a ...int) int {
	s := 0
	for _, x := range a {
		s += x
	}
	return s
}

// truncated
func Divmod(a, b int) (int, int) {
	q := a / b
	r := a - q*b
	return q, r
}

func Lb(a []int, x int) int {
	n := len(a)
	f := func(i int) bool {
		return a[i] >= x
	}
	return sort.Search(n, f)
}

func Ub(a []int, x int) int {
	n := len(a)
	f := func(i int) bool {
		return a[i] > x
	}
	return sort.Search(n, f)
}

func main() {
	defer wr.Flush()
	t := 1
	// t := Int()
	for ; t > 0; t-- {
		Solve()
	}
}

func Iota(n int) []int {
	a := make([]int, n)
	for i := 0; i < n; i++ {
		a[i] = i
	}
	return a
}

func Egcd(a, b int) (int, int, int) {
	if b == 0 {
		if a < 0 {
			return -a, -1, 0
		} else {
			return a, 1, 0
		}
	}
	q, r := Divmod(a, b)
	g, s, t := Egcd(b, r)
	return g, t, s - q*t
}

func Inv(m, a int) int {
	Assert(0 < a && a < m)
	g, x, _ := Egcd(a, m)
	Assert(g == 1)
	if x < 0 {
		x += m
	}
	return x
}

var Mod int

func SetMod(m int) {
	Assert(m > 0)
	Mod = m
}

var Fact, IFact, Invs []int

func InitTables(n int) {
	Fact = make([]int, n)
	IFact = make([]int, n)
	Invs = make([]int, n)
	Fact[0] = 1
	IFact[0] = 1
	for i := 1; i < n; i++ {
		Fact[i] = Fact[i-1] * i % Mod
	}
	IFact[n-1] = Inv(Mod, Fact[n-1])
	for i := n - 1; i > 0; i-- {
		IFact[i-1] = IFact[i] * i % Mod
		Invs[i] = IFact[i] * Fact[i-1] % Mod
	}
}

func Perm(n, k int) int {
	if k < 0 || n < k {
		return 0
	}
	return Fact[n] * IFact[n-k] % Mod

}

func Choose(n, k int) int {
	if k < 0 {
		return 0
	}
	return Perm(n, k) * IFact[k] % Mod
}

func Solve() {
	const mod = 998_244_353
	SetMod(mod)
	x, y := Int(), Int()
	z, w := Int(), Int()

	Chmax(&z, 1)
	Chmax(&w, 1)

	InitTables(1 << 20)
	ans := Fact[x+y-z-w]
	ans *= Choose(x, z)
	ans %= mod
	ans *= Choose(y, w)
	ans %= mod
	Out(ans)
}
0