程序员求职经验分享与学习资料整理平台

网站首页 > 文章精选 正文

2025-07-05:统计异或值为给定值的路径数目。用go语言,给定一个

balukai 2025-07-14 14:49:00 文章精选 4 ℃

2025-07-05:统计异或值为给定值的路径数目。用go语言,给定一个大小为 m 行 n 列的二维整数数组 grid 和一个整数 k。

任务是计算从左上角起点 (0, 0) 出发,到右下角终点 (m-1, n-1) 的所有路径数量,这些路径必须满足以下条件:

  • o 每一步只能向右或向下移动(即从 (i, j) 到 (i, j+1) 或 (i+1, j))。
  • o 路径上经过的所有数字进行异或(XOR)运算后的结果等于 k。

由于结果可能很大,需要将最终路径数对 1000000007 取模后返回。

1 <= m == grid.length <= 300。

1 <= n == grid[r].length <= 300。

0 <= grid[r][c] < 16。

0 <= k < 16。

输入:grid = [[2, 1, 5], [7, 10, 0], [12, 6, 4]], k = 11。

输出:3。

解释:

3 条路径分别为:

(0, 0) → (1, 0) → (2, 0) → (2, 1) → (2, 2)。

(0, 0) → (1, 0) → (1, 1) → (1, 2) → (2, 2)。

(0, 0) → (0, 1) → (1, 1) → (2, 1) → (2, 2)。

题目来自力扣3393。

解决思路

这是一个典型的动态规划问题。我们需要跟踪从起点到每个单元格的所有可能的异或值及其对应的路径数量。

关键点:

  1. 1. 异或运算的性质:异或运算是可逆的,且顺序不影响最终结果。即 a XOR b XOR c = a XOR (b XOR c)
  2. 2. 动态规划状态定义:我们可以定义 f[i][j][x] 表示从起点 (0, 0) 到 (i, j) 的所有路径中,异或结果为 x 的路径数量。
  3. 3. 状态转移:对于每个单元格 (i, j),其异或值 x 可以通过从上方 (i-1, j) 或左方 (i, j-1) 的单元格转移而来:
  4. o 如果从上方来,则 x = x_above XOR grid[i][j]
  5. o 如果从左方来,则 x = x_left XOR grid[i][j]
  6. o 因此,f[i][j][x] = f[i-1][j][x XOR grid[i][j]] + f[i][j-1][x XOR grid[i][j]]
  7. 4. 初始条件:起点的异或值是 grid[0][0],因此 f[0][0][grid[0][0]] = 1

具体步骤:

  1. 1. 确定异或值的上限
  2. o 网格中的最大值决定了异或结果的最大可能值。因为 0 <= grid[r][c] < 16,所以异或结果的最大值是 15(二进制 1111),因此异或值的范围是 0 到 15。
  3. o 代码中通过 bits.Len(uint(mx)) 计算最大值的二进制位数,然后 1 << bits.Len(uint(mx)) 得到上限 u。由于 mx 最大是 15,u 是 16。
  4. o 如果 k >= u,直接返回 0,因为异或结果不可能大于等于 u
  5. 2. 初始化动态规划表
  6. o 创建一个三维数组 f,大小为 (m+1) x (n+1) x u,初始化为 0。
  7. o 为了方便边界处理,代码中从 (1, 1) 开始对应网格的 (0, 0),因此 f[0][1][0] = 1f[1][0][0] = 1 是初始条件。这相当于从虚拟的起点出发,异或值为 0。
  8. 3. 填充动态规划表
  9. o 遍历网格的每个单元格 (i, j)。
  10. o 对于每个可能的异或值 x(从 0 到 u-1),计算从上方和左方转移来的路径数量:
  11. o f[i+1][j+1][x] = (f[i+1][j][x XOR val] + f[i][j+1][x XOR val]) % mod,其中 val = grid[i][j]
  12. o 这样,f[i+1][j+1][x] 表示到 (i, j) 时异或值为 x 的路径数量。
  13. 4. 返回结果
  14. o 最终结果是 f[m][n][k],即从 (0, 0) 到 (m-1, n-1) 异或值为 k 的路径数量。

时间复杂度和空间复杂度

  • o 时间复杂度
    • o 我们需要遍历每个单元格 (i, j),共 m x n 个。
    • o 对于每个单元格,我们需要遍历所有可能的异或值 x,最多 16 种。
    • o 因此,时间复杂度是 O(m * n * 16),即 O(m * n),因为 16 是常数。
  • o 空间复杂度
    • o 动态规划表 f 的大小是 (m+1) x (n+1) x 16
    • o 因此,空间复杂度是 O(m * n * 16),即 O(m * n)

总结

通过动态规划方法,我们高效地统计了满足条件的路径数量。时间和空间复杂度都是网格大小的线性倍数(乘以常数 16),适用于给定的约束条件(m, n <= 300)。

Go完整代码如下:

.

package main

import (
    "fmt"
    "slices"
    "math/bits"
)

func countPathsWithXorValue(grid [][]int, k int)int {
    const mod = 1_000_000_007
    mx := 0
    for _, row := range grid {
        mx = max(mx, slices.Max(row))
    }
    u := 1 << bits.Len(uint(mx))
    if k >= u {
        return0
    }

    m, n := len(grid), len(grid[0])
    f := make([][][]int, m+1)
    for i := range f {
        f[i] = make([][]int, n+1)
        for j := range f[i] {
            f[i][j] = make([]int, u)
        }
    }
    f[0][1][0] = 1
    for i, row := range grid {
        for j, val := range row {
            for x := range u {
                f[i+1][j+1][x] = (f[i+1][j][x^val] + f[i][j+1][x^val]) % mod
            }
        }
    }
    return f[m][n][k]
}

func main() {
    grid := [][]int{{2, 1, 5}, {7, 10, 0}, {12, 6, 4}}
    k := 11
    result := countPathsWithXorValue(grid,k)
    fmt.Println(result)
}


Python完整代码如下:

.

# -*-coding:utf-8-*-

import math

def countPathsWithXorValue(grid, k):
    mod = 1_000_000_007
    
    mx = 0
    for row in grid:
        mx = max(mx, max(row))
    u = 1 << (mx.bit_length())
    if k >= u:
        return0

    m, n = len(grid), len(grid[0])
    # 初始化三维dp数组 f[m+1][n+1][u]
    f = [[[0] * u for _ in range(n+1)] for __ in range(m+1)]
    f[0][1][0] = 1

    for i in range(m):
        for j in range(n):
            val = grid[i][j]
            for x in range(u):
                # 从左边(f[i+1][j])和上边(f[i][j+1])转移
                f[i+1][j+1][x] = (f[i+1][j][x ^ val] + f[i][j+1][x ^ val]) % mod

    return f[m][n][k]

if __name__ == "__main__":
    grid = [[2, 1, 5], [7, 10, 0], [12, 6, 4]]
    k = 11
    result = countPathsWithXorValue(grid, k)
    print(result)



·



我们相信Go语言和算法为普通开发者提供了困境的“面试利器”,并致力于分享全面的编程知识。在这里,您可以找到最新的Go语言教程、算法解析、提升面试竞争力的秘籍以及行业动态。


欢迎关注“福大规模架构师每日一题”,让 Go 语言和算法助力您的职业发展

·

Tags:

最近发表
标签列表