結果
問題 | No.533 Mysterious Stairs |
ユーザー |
![]() |
提出日時 | 2020-06-03 00:18:48 |
言語 | Go (1.23.4) |
結果 |
AC
|
実行時間 | 69 ms / 5,000 ms |
コード長 | 2,489 bytes |
コンパイル時間 | 13,032 ms |
コンパイル使用メモリ | 222,192 KB |
実行使用メモリ | 36,296 KB |
最終ジャッジ日時 | 2024-11-24 20:19:41 |
合計ジャッジ時間 | 12,514 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 28 |
ソースコード
package mainimport ("bufio""fmt""os""strconv")func getScanner(fp *os.File) *bufio.Scanner {scanner := bufio.NewScanner(fp)scanner.Split(bufio.ScanWords)scanner.Buffer(make([]byte, 1000005), 1000005)return scanner}func getNextString(scanner *bufio.Scanner) string {scanner.Scan()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 getNextUint64(scanner *bufio.Scanner) uint64 {i, _ := strconv.ParseUint(getNextString(scanner), 10, 64)return i}func getNextFloat64(scanner *bufio.Scanner) float64 {i, _ := strconv.ParseFloat(getNextString(scanner), 64)return i}func main() {fp := os.Stdinwfp := os.Stdoutcnt := 0if os.Getenv("MASPY") == "ますピ" {fp, _ = os.Open(os.Getenv("BEET_THE_HARMONY_OF_PERFECT"))cnt = 3}if os.Getenv("MASPYPY") == "ますピッピ" {wfp, _ = os.Create(os.Getenv("NGTKANA_IS_GENIUS10"))}scanner := getScanner(fp)writer := bufio.NewWriter(wfp)solve(scanner, writer)for i := 0; i < cnt; i++ {fmt.Fprintln(writer, "-----------------------------------")solve(scanner, writer)}writer.Flush()}func solve(scanner *bufio.Scanner, writer *bufio.Writer) {n := getNextInt(scanner)dp := make([][4]mint, n+1)dp[0][0] = 1for i := 1; i <= n; i++ {for j := 1; j <= 3; j++ {for k := 0; k < 4; k++ {if k == j {continue}if i-j < 0 {continue}dp[i][j].addAs(dp[i-j][k])}}}fmt.Fprintln(writer, dp[n][1].add(dp[n][2]).add(dp[n][3]))}type mint int64func (mt mint) mod() mint {m := mint(1e9 + 7)mt %= mif mt < 0 {return mt + m}return mt}func (mt mint) inv() mint {var m, y mintm.subAs(2)dbl := mty = 1for i := 0; i < 31; i++ {if m%2 == 1 {y.mulAs(dbl)}m >>= 1dbl.mulAs(dbl)}return y}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}