結果

問題 No.117 組み合わせの数
ユーザー fmhrfmhr
提出日時 2016-09-10 09:33:06
言語 Go
(1.22.1)
結果
AC  
実行時間 318 ms / 5,000 ms
コード長 1,866 bytes
コンパイル時間 11,201 ms
コンパイル使用メモリ 211,788 KB
実行使用メモリ 54,812 KB
最終ジャッジ日時 2023-08-10 13:24:33
合計ジャッジ時間 12,288 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 318 ms
54,812 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

package main

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

func main() {
	sc.Split(bufio.ScanWords)
	T := nextInt()
	initCombination()
	for i := 0; i < T; i++ {
		S := nextLine()
		ai, _ := strconv.Atoi(strings.Split(S[2:len(S)-1], ",")[0])
		bi, _ := strconv.Atoi(strings.Split(S[2:len(S)-1], ",")[1])
		a := uint64(ai)
		b := uint64(bi)
		var ans uint64
		switch S[0] {
		case 'C':
			ans = c(a, b)
		case 'P':
			ans = p(a, b)
		case 'H':
			ans = h(a, b)
		}
		fmt.Println(ans)
	}

}

const MAX = 2000010

var MOD uint64 = 1000000007 // 素数
var fact [MAX]uint64        // 階乗
var inv [MAX]uint64
var factr [MAX]uint64 // 階乗の逆元
func initCombination() {
	inv[1], fact[0], factr[0] = 1, 1, 1
	// nHkはn+k+1Ckとなるためn+kをとる
	var i uint64
	for i = 2; i < MAX; i++ {
		inv[i] = inv[MOD%i] * (MOD - MOD/i) % MOD
	}
	for i = 1; i < MAX; i++ {
		fact[i] = fact[i-1] * i % MOD
		factr[i] = factr[i-1] * inv[i] % MOD
	}
}

// 1 以上 N 以下の N 個の整数の中から，相異なる K 個の整数を選ぶパターンの数を C(N,K) と書きます．
func c(n, c uint64) uint64 {
	if c < 0 || c > n {
		return 0
	}
	return factr[c] * fact[n] % MOD * factr[n-c] % MOD
}

// 1 以上 N 以下の N 個の整数の中から，相異なる K 個の整数を選び，順番に並べるパターンの数を P(N,K) と書きます．
func p(n, c uint64) uint64 {
	if c < 0 || c > n {
		return 0
	}
	return fact[n] * factr[n-c] % MOD
}

// 1 以上 N 以下の N 個の整数の中から，重複を許して K 個の整数を選ぶパターンの数を H(N,K) と書きます．
func h(p, q uint64) uint64 {
	if p == 0 && q == 0 {
		return 1
	}
	return c(p+q-1, q)
}

var sc = bufio.NewScanner(os.Stdin)

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

func nextInt() int {
	i, _ := strconv.Atoi(nextLine())
	return i
}
0