結果
問題 | No.978 Fibonacci Convolution Easy |
ユーザー | ccppjsrb |
提出日時 | 2020-08-26 17:42:50 |
言語 | Go (1.22.1) |
結果 |
AC
|
実行時間 | 65 ms / 2,000 ms |
コード長 | 2,421 bytes |
コンパイル時間 | 14,169 ms |
コンパイル使用メモリ | 228,208 KB |
実行使用メモリ | 33,024 KB |
最終ジャッジ日時 | 2024-04-25 03:31:20 |
合計ジャッジ時間 | 15,227 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 27 ms
13,952 KB |
testcase_02 | AC | 15 ms
7,936 KB |
testcase_03 | AC | 61 ms
31,744 KB |
testcase_04 | AC | 18 ms
9,344 KB |
testcase_05 | AC | 5 ms
5,376 KB |
testcase_06 | AC | 24 ms
12,288 KB |
testcase_07 | AC | 41 ms
21,120 KB |
testcase_08 | AC | 30 ms
15,104 KB |
testcase_09 | AC | 47 ms
24,064 KB |
testcase_10 | AC | 65 ms
33,024 KB |
testcase_11 | AC | 21 ms
10,624 KB |
testcase_12 | AC | 4 ms
5,376 KB |
testcase_13 | AC | 23 ms
12,032 KB |
testcase_14 | AC | 8 ms
5,376 KB |
testcase_15 | AC | 26 ms
13,568 KB |
testcase_16 | AC | 65 ms
33,024 KB |
testcase_17 | AC | 65 ms
32,896 KB |
testcase_18 | AC | 1 ms
5,376 KB |
testcase_19 | AC | 1 ms
5,376 KB |
testcase_20 | AC | 2 ms
5,376 KB |
ソースコード
package main import ( "bufio" "fmt" "os" "strconv" ) func configure(scanner *bufio.Scanner) { scanner.Split(bufio.ScanWords) scanner.Buffer(make([]byte, 1000005), 1000005) } func getNextString(scanner *bufio.Scanner) string { scanned := scanner.Scan() if !scanned { panic("scan failed") } return scanner.Text() } func getNextInt(scanner *bufio.Scanner) int { i, _ := strconv.Atoi(getNextString(scanner)) return i } func getNextInt64(scanner *bufio.Scanner) int64 { i, _ := strconv.ParseInt(getNextString(scanner), 10, 64) return i } func getNextFloat64(scanner *bufio.Scanner) float64 { i, _ := strconv.ParseFloat(getNextString(scanner), 64) return i } func main() { fp := os.Stdin wfp := os.Stdout extra := 0 if os.Getenv("I") == "IronMan" { fp, _ = os.Open(os.Getenv("END_GAME")) extra = 100 } scanner := bufio.NewScanner(fp) configure(scanner) writer := bufio.NewWriter(wfp) defer func() { r := recover() if r != nil { fmt.Fprintln(writer, r) } writer.Flush() }() solve(scanner, writer) for i := 0; i < extra; i++ { fmt.Fprintln(writer, "-----------------------------------") solve(scanner, writer) } } func solve(scanner *bufio.Scanner, writer *bufio.Writer) { n := getNextInt(scanner) p := mint(getNextInt(scanner)) ff := make([]mint, n+5) ss := make([]mint, n+5) ff[1] = 0 ff[2] = 1 for i := 3; i < n+1; i++ { ff[i] = ff[i-1].mul(p).add(ff[i-2]) } for i := 1; i < n+2; i++ { ss[i] = ss[i-1].add(ff[i-1]) } var ans mint for i := 1; i < n+1; i++ { ans.addAs(ff[i].mul(ss[n+1].sub(ss[i]))) } fmt.Fprintln(writer, ans) } type mint int64 func (mt mint) mod() mint { m := mint(1e9 + 7) mt %= m if mt < 0 { return mt + m } return mt } func (mt mint) inv() mint { return mt.pow(mint(0).sub(2)) } func (mt mint) pow(n mint) mint { p := mint(1) for n > 0 { if n&1 == 1 { p.mulAs(mt) } mt.mulAs(mt) n >>= 1 } return p } func (mt mint) add(x mint) mint { return (mt + x).mod() } func (mt mint) sub(x mint) mint { return (mt - x).mod() } func (mt mint) mul(x mint) mint { return (mt * x).mod() } func (mt mint) div(x mint) mint { return mt.mul(x.inv()) } func (mt *mint) addAs(x mint) *mint { *mt = mt.add(x) return mt } func (mt *mint) subAs(x mint) *mint { *mt = mt.sub(x) return mt } func (mt *mint) mulAs(x mint) *mint { *mt = mt.mul(x) return mt } func (mt *mint) divAs(x mint) *mint { *mt = mt.div(x) return mt }