結果
問題 |
No.1967 Sugoroku Optimization
|
ユーザー |
|
提出日時 | 2022-06-26 16:49:57 |
言語 | Go (1.23.4) |
結果 |
AC
|
実行時間 | 725 ms / 2,000 ms |
コード長 | 1,079 bytes |
コンパイル時間 | 14,152 ms |
コンパイル使用メモリ | 217,884 KB |
実行使用メモリ | 34,432 KB |
最終ジャッジ日時 | 2024-11-17 02:41:41 |
合計ジャッジ時間 | 20,799 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
package main import ( "bufio" "fmt" "os" "strconv" ) var sc = bufio.NewScanner(os.Stdin) var out = bufio.NewWriter(os.Stdout) func solve(n, k int) int { const p = 998244353 //jの位置にいる時、残りi回でNに辿り着ける確率 dp := make([][]int, k+1) for i := 0; i <= k; i++ { dp[i] = make([]int, n+1) } for i := 0; i <= k; i++ { dp[i][n] = 1 } for i := 0; i < k; i++ { var s, inv int for j := n - 1; j >= 0; j-- { s += dp[i][j+1] s %= p inv++ inv %= p dp[i+1][j] = s * Inv(inv, p) % p } } return dp[k][0] } func main() { buf := make([]byte, 1024*1024) sc.Buffer(buf, bufio.MaxScanTokenSize) sc.Split(bufio.ScanWords) n, k := nextInt(), nextInt() ans := solve(n, k) PrintInt(ans) } func nextInt() int { sc.Scan() i, _ := strconv.Atoi(sc.Text()) return i } func PrintInt(x int) { defer out.Flush() fmt.Fprintln(out, x) } func Pow(x, y, p int) int { ret := 1 for y > 0 { if y%2 == 1 { ret = ret * x % p } y >>= 1 x = x * x % p } return ret } func Inv(x, p int) int { return Pow(x, p-2, p) }