結果
問題 | No.2003 Frog on Grid |
ユーザー | 草苺奶昔 |
提出日時 | 2023-05-12 23:16:30 |
言語 | Go (1.22.1) |
結果 |
AC
|
実行時間 | 423 ms / 2,000 ms |
コード長 | 3,061 bytes |
コンパイル時間 | 12,191 ms |
コンパイル使用メモリ | 228,096 KB |
実行使用メモリ | 56,960 KB |
最終ジャッジ日時 | 2024-05-06 13:17:08 |
合計ジャッジ時間 | 16,765 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 1 ms
5,248 KB |
testcase_02 | AC | 1 ms
5,248 KB |
testcase_03 | AC | 1 ms
5,376 KB |
testcase_04 | AC | 1 ms
5,376 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 | 2 ms
5,376 KB |
testcase_10 | AC | 1 ms
5,376 KB |
testcase_11 | AC | 1 ms
5,376 KB |
testcase_12 | AC | 1 ms
5,376 KB |
testcase_13 | AC | 2 ms
5,376 KB |
testcase_14 | AC | 1 ms
5,376 KB |
testcase_15 | AC | 1 ms
5,376 KB |
testcase_16 | AC | 1 ms
5,376 KB |
testcase_17 | AC | 71 ms
7,936 KB |
testcase_18 | AC | 256 ms
19,200 KB |
testcase_19 | AC | 172 ms
21,760 KB |
testcase_20 | AC | 423 ms
56,960 KB |
testcase_21 | AC | 131 ms
21,760 KB |
testcase_22 | AC | 391 ms
56,960 KB |
testcase_23 | AC | 408 ms
19,200 KB |
testcase_24 | AC | 407 ms
18,944 KB |
testcase_25 | AC | 275 ms
19,072 KB |
testcase_26 | AC | 280 ms
19,072 KB |
testcase_27 | AC | 43 ms
19,072 KB |
testcase_28 | AC | 38 ms
19,072 KB |
ソースコード
package main import ( "bufio" "fmt" "os" ) // https://yukicoder.me/problems/no/2003 // 青蛙从(0,0)出发 可以跳到1<=dx+dy<=k的格子上 // 不能跳到'#'上 // !求跳到(n-1,m-1)的方案数 // 2<=n*m<=1e6 func main() { in := bufio.NewReader(os.Stdin) out := bufio.NewWriter(os.Stdout) defer out.Flush() var ROW, COL, k int fmt.Fscan(in, &ROW, &COL, &k) grid := make([]string, ROW) for i := 0; i < ROW; i++ { fmt.Fscan(in, &grid[i]) } bit2 := NewFenwickTree2DDense(ROW, COL) dp := make([][]int, ROW) for i := 0; i < ROW; i++ { dp[i] = make([]int, COL) } dp[0][0] = 1 bit2.Add(0, 0, 1) // x+y=s getRange := func(s int) (int, int) { lo := max(0, s-COL+1) hi := min(ROW-1, s) return lo, hi + 1 } for sum := 1; sum < ROW+COL-1; sum++ { a, b := getRange(sum - k - 1) for x := a; x < b; x++ { y := sum - k - 1 - x if grid[x][y] == '#' { continue } bit2.Add(x, y, -dp[x][y]) } a, b = getRange(sum) for x := a; x < b; x++ { y := sum - x if grid[x][y] == '#' { continue } dp[x][y] = bit2.QueryPrefix(x+1, y+1) bit2.Add(x, y, dp[x][y]) } } fmt.Fprintln(out, dp[ROW-1][COL-1]) } const MOD int = 998244353 type E = int func e() E { return 0 } func op(a, b E) E { res := (a + b) % MOD if res < 0 { res += MOD } return res } func inv(a E) E { return -a } type FenwickTree2DDense struct { H, W int data []E unit E } // 指定二维树状数组的高度和宽度. func NewFenwickTree2DDense(h, w int) *FenwickTree2DDense { res := &FenwickTree2DDense{H: h, W: w, data: make([]E, h*w), unit: e()} for i := 0; i < h*w; i++ { res.data[i] = res.unit } return res } // 点 (x,y) 的值加上 val. func (ft *FenwickTree2DDense) Add(x, y int, val E) { x++ for x <= ft.H { ft.addX(x, y, val) x += x & -x } } // [lx,rx) * [ly,ry) func (ft *FenwickTree2DDense) Query(lx, rx, ly, ry int) E { pos, neg := ft.unit, ft.unit for lx < rx { pos = op(pos, ft.sumX(rx, ly, ry)) rx -= rx & -rx } for rx < lx { neg = op(neg, ft.sumX(lx, ly, ry)) lx -= lx & -lx } return op(pos, inv(neg)) } // [0,rx) * [0,ry) func (ft *FenwickTree2DDense) QueryPrefix(rx, ry int) E { pos := ft.unit for rx > 0 { pos = op(pos, ft.sumXPrefix(rx, ry)) rx -= rx & -rx } return pos } func (ft *FenwickTree2DDense) idx(x, y int) int { return ft.W*(x-1) + (y - 1) } func (ft *FenwickTree2DDense) addX(x, y int, val E) { y++ for y <= ft.W { ft.data[ft.idx(x, y)] = op(ft.data[ft.idx(x, y)], val) y += y & -y } } func (ft *FenwickTree2DDense) sumX(x, ly, ry int) E { pos, neg := ft.unit, ft.unit for ly < ry { pos = op(pos, ft.data[ft.idx(x, ry)]) ry -= ry & -ry } for ry < ly { neg = op(neg, ft.data[ft.idx(x, ly)]) ly -= ly & -ly } return op(pos, inv(neg)) } func (ft *FenwickTree2DDense) sumXPrefix(x, ry int) E { pos := ft.unit for ry > 0 { pos = op(pos, ft.data[ft.idx(x, ry)]) ry -= ry & -ry } return pos } func min(a, b int) int { if a < b { return a } return b } func max(a, b int) int { if a > b { return a } return b }