#568799 (Go) No.389 ロジックパズルの組み合わせ

提出ソース

結果

問題	No.389 ロジックパズルの組み合わせ
ユーザー	aru aru
提出日時	2020-10-18 20:49:34
言語	Go (1.22.1)
結果	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

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	1 ms 5,376 KB
testcase_03	AC	1 ms 5,376 KB
testcase_04	AC	225 ms 25,040 KB
testcase_05	AC	1 ms 5,376 KB
testcase_06	AC	1 ms 5,376 KB
testcase_07	AC	1 ms 5,376 KB
testcase_08	AC	1 ms 5,376 KB
testcase_09	AC	229 ms 30,328 KB
testcase_10	AC	24 ms 6,272 KB
testcase_11	AC	227 ms 27,812 KB
testcase_12	AC	11 ms 5,376 KB
testcase_13	AC	72 ms 12,160 KB
testcase_14	AC	23 ms 6,272 KB
testcase_15	AC	33 ms 7,040 KB
testcase_16	AC	99 ms 15,292 KB
testcase_17	AC	148 ms 20,736 KB
testcase_18	AC	134 ms 20,260 KB
testcase_19	AC	1 ms 5,376 KB
testcase_20	AC	1 ms 5,376 KB
testcase_21	AC	2 ms 5,376 KB
testcase_22	AC	2 ms 5,376 KB
testcase_23	AC	1 ms 5,376 KB
testcase_24	AC	1 ms 5,376 KB
testcase_25	AC	2 ms 5,376 KB
testcase_26	AC	1 ms 5,376 KB
testcase_27	AC	1 ms 5,376 KB
testcase_28	AC	1 ms 5,376 KB
testcase_29	AC	1 ms 5,376 KB
testcase_30	AC	1 ms 5,376 KB
testcase_31	AC	1 ms 5,376 KB
testcase_32	AC	1 ms 5,376 KB
testcase_33	AC	1 ms 5,376 KB
testcase_34	AC	1 ms 5,376 KB
testcase_35	AC	1 ms 5,376 KB
testcase_36	AC	1 ms 5,376 KB
testcase_37	AC	1 ms 5,376 KB
testcase_38	AC	1 ms 5,376 KB
testcase_39	AC	1 ms 5,376 KB
testcase_40	AC	1 ms 5,376 KB
testcase_41	AC	1 ms 5,376 KB
testcase_42	AC	1 ms 5,376 KB
testcase_43	AC	2 ms 5,376 KB
testcase_44	AC	2 ms 5,376 KB
testcase_45	AC	1 ms 5,376 KB
testcase_46	AC	2 ms 5,376 KB
testcase_47	AC	1 ms 5,376 KB
testcase_48	AC	1 ms 5,376 KB
testcase_49	AC	29 ms 19,072 KB
testcase_50	AC	2 ms 5,376 KB
testcase_51	AC	2 ms 5,376 KB
testcase_52	AC	4 ms 5,376 KB
testcase_53	AC	3 ms 5,376 KB
testcase_54	AC	5 ms 5,376 KB
testcase_55	AC	2 ms 5,376 KB
testcase_56	AC	3 ms 5,376 KB
testcase_57	AC	9 ms 9,728 KB
testcase_58	AC	24 ms 15,744 KB
testcase_59	AC	18 ms 11,264 KB
testcase_60	AC	7 ms 7,040 KB
testcase_61	AC	31 ms 28,160 KB
testcase_62	AC	8 ms 7,424 KB
testcase_63	AC	26 ms 16,128 KB
testcase_64	AC	6 ms 5,376 KB
testcase_65	AC	3 ms 5,376 KB
testcase_66	AC	6 ms 5,632 KB
testcase_67	AC	6 ms 5,504 KB
testcase_68	AC	6 ms 5,376 KB
testcase_69	AC	124 ms 17,988 KB
testcase_70	AC	57 ms 9,472 KB
testcase_71	AC	90 ms 13,568 KB
testcase_72	AC	37 ms 6,912 KB
testcase_73	AC	68 ms 10,880 KB
testcase_74	AC	97 ms 14,976 KB
testcase_75	AC	85 ms 13,568 KB
testcase_76	AC	30 ms 6,144 KB
testcase_77	AC	169 ms 26,220 KB
testcase_78	AC	10 ms 5,376 KB
testcase_79	AC	1 ms 5,376 KB
testcase_80	AC	1 ms 5,376 KB
testcase_81	AC	1 ms 5,376 KB
testcase_82	AC	1 ms 5,376 KB
testcase_83	AC	1 ms 5,376 KB
testcase_84	AC	1 ms 5,376 KB
testcase_85	AC	1 ms 5,376 KB
testcase_86	AC	1 ms 5,376 KB
testcase_87	AC	1 ms 5,376 KB
testcase_88	AC	1 ms 5,376 KB
testcase_89	AC	44 ms 8,064 KB
testcase_90	AC	78 ms 11,520 KB
testcase_91	AC	139 ms 19,840 KB
testcase_92	AC	19 ms 5,888 KB
testcase_93	AC	86 ms 13,184 KB
testcase_94	AC	3 ms 5,376 KB
testcase_95	AC	92 ms 12,280 KB
testcase_96	AC	69 ms 9,728 KB
testcase_97	AC	65 ms 11,136 KB
testcase_98	AC	28 ms 6,016 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

package main

import (
	"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 := 0
	for _, e := range s {
		v, _ := strconv.Atoi(e)
		h = append(h, v)
		tot += v
	}
	if tot == 0 {
		out(1)
		return
	}
	tot += len(h) - 1
	m := M - tot
	n := len(h)

	if m < 0 {
		out("NA")
		return
	}

	out(nCr(n+m, m))
}

const mod = 1000000007

func 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 := 1
	ret *= mfrac(n)
	ret %= mod
	ret *= mifrac(r)
	ret %= mod
	ret *= mifrac(n - r)
	ret %= mod
	return (ret)
}

yukicoder

結果

テストケース

ソースコード