結果
問題 | No.389 ロジックパズルの組み合わせ |
ユーザー |
![]() |
提出日時 | 2020-10-18 20:49:34 |
言語 | Go (1.23.4) |
結果 |
AC
|
実行時間 | 229 ms / 2,000 ms |
コード長 | 2,609 bytes |
コンパイル時間 | 15,773 ms |
コンパイル使用メモリ | 219,300 KB |
実行使用メモリ | 30,328 KB |
最終ジャッジ日時 | 2024-07-21 06:43:31 |
合計ジャッジ時間 | 21,088 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 99 |
ソースコード
package mainimport ("bufio""fmt""os""sort""strconv""strings")var sc = bufio.NewScanner(os.Stdin)var wr = bufio.NewWriter(os.Stdout)func out(x ...interface{}) {fmt.Fprintln(wr, x...)}func getI() int {sc.Scan()i, e := strconv.Atoi(sc.Text())if e != nil {panic(e)}return i}func getF() float64 {sc.Scan()i, e := strconv.ParseFloat(sc.Text(), 64)if e != nil {panic(e)}return i}func getInts(N int) []int {ret := make([]int, N)for i := 0; i < N; i++ {ret[i] = getI()}return ret}func getS() string {sc.Scan()return sc.Text()}// min, max, asub, absなど基本関数func max(a, b int) int {if a > b {return a}return b}func min(a, b int) int {if a < b {return a}return b}func asub(a, b int) int {if a > b {return a - b}return b - a}func abs(a int) int {if a >= 0 {return a}return -a}func lowerBound(a []int, x int) int {idx := sort.Search(len(a), func(i int) bool {return a[i] >= x})return idx}func upperBound(a []int, x int) int {idx := sort.Search(len(a), func(i int) bool {return a[i] > x})return idx}func main() {defer wr.Flush()sc.Split(bufio.ScanLines)sc.Buffer([]byte{}, 1000000)// this template is new version.// use getI(), getS(), getInts(), getF()M := getI()s := strings.Split(getS(), " ")h := make([]int, 0)tot := 0for _, e := range s {v, _ := strconv.Atoi(e)h = append(h, v)tot += v}if tot == 0 {out(1)return}tot += len(h) - 1m := M - totn := len(h)if m < 0 {out("NA")return}out(nCr(n+m, m))}const mod = 1000000007func mpow(a, b int) int {if b == 0 {return 1} else if b%2 == 0 {tmp := mpow(a, b/2)return tmp * tmp % mod}return mpow(a, b-1) * a % mod}var fracMemo = []int{1, 1}func mfrac(n int) int {if len(fracMemo) > n {return fracMemo[n]}if len(fracMemo) == 0 {fracMemo = append(fracMemo, 1)}for len(fracMemo) <= n {size := len(fracMemo)fracMemo = append(fracMemo, fracMemo[size-1]*size%mod)}return fracMemo[n]}var ifracMemo = []int{1, 1}func mifrac(n int) int {if len(ifracMemo) > n {return ifracMemo[n]}if len(ifracMemo) == 0 {fracMemo = append(ifracMemo, 1)}for len(ifracMemo) <= n {size := len(ifracMemo)ifracMemo = append(ifracMemo, ifracMemo[size-1]*mpow(size, mod-2)%mod)}return ifracMemo[n]}func nCr(n, r int) int {if n == r {return 1}if n < r || r < 0 {return 0}ret := 1ret *= mfrac(n)ret %= modret *= mifrac(r)ret %= modret *= mifrac(n - r)ret %= modreturn (ret)}