2023-03-26:给定一个二维数组matrix, 每个格子都是正数,每个格子都和上、下、左、右相邻

2023-03-26:给定一个二维数组matrix,
每个格子都是正数,每个格子都和上、下、左、右相邻。
你可以从任何一个格子出发,走向相邻的格子,
把沿途的数字乘起来,希望得到的最终数字中,结尾的0最多,
走的过程中,向左走或者向右走的拐点,最多只能有一次。
返回结尾最多的0,能是多少。
1 <= 行、列 <= 400。

答案2023-03-26:

解题思路

本题需要求出从任意位置出发,最多能有多少个结尾0。为了方便计算,可以先将矩阵中每个数分解成2和5的因子,然后通过前缀和预处理出每个位置上、左方向的2和5的因子数量之和,以便快速计算6个方向上的因子数量之和。接着遍历每个位置,分别计算6个方向上的因子数量之和,并取其中的最小值,最后返回所有最小值中的最大值即可。

具体来说,对于一个位置(i,j),可以计算它的左、右、上、下4个方向的2和5的因子数量之和,以及两个斜方向的2和5的因子数量之和共6个值。然后取这6个值中的最小值,就是从该位置出发,在该方向上能够得到的最多结尾0的数量。遍历所有位置,取最大值即可。

需要注意的是,由于只能有一次向左或向右的拐点,因此在计算左和右方向上的因子数量之和时,需要分别计算到该行左边界和右边界的因子数量之和,然后再通过减法计算出中间部分的因子数量之和。

时间复杂度

本算法需要对矩阵中每个数进行分解质因数,时间复杂度为O(n^2log(max(matrix)));两层循环分别对n和m进行遍历,时间复杂度为O(nm);因此总时间复杂度为O(n^2log(max(matrix))+nm)。

空间复杂度

本算法需要维护4个二维数组,每个数组的大小均为n×m,因此空间复杂度为O(nm)。

rust代码

fn most_trailing_zeros(matrix: &Vec<Vec<i32>>) -> i32 {
    let n = matrix.len(); // 矩阵行数
    let m = matrix[0].len(); // 矩阵列数

    // f2[i][j] : matrix[i][j]自己有几个2的因子
    let mut f2 = vec![vec![0; m]; n];
    // f5[i][j] : matrix[i][j]自己有几个5的因子
    let mut f5 = vec![vec![0; m]; n];

    // 预处理每个位置的2和5的因子数量
    for i in 0..n {
        for j in 0..m {
            f2[i][j] = factors(matrix[i][j], 2);
            f5[i][j] = factors(matrix[i][j], 5);
        }
    }

    // 计算每个位置上、左方向的2和5的因子数量之和
    let mut left_f2 = vec![vec![0; m]; n];
    let mut left_f5 = vec![vec![0; m]; n];
    let mut up_f2 = vec![vec![0; m]; n];
    let mut up_f5 = vec![vec![0; m]; n];

    for i in 0..n {
        for j in 0..m {
            left_f2[i][j] = f2[i][j] + if j > 0 { left_f2[i][j - 1] } else { 0 };
            left_f5[i][j] = f5[i][j] + if j > 0 { left_f5[i][j - 1] } else { 0 };
            up_f2[i][j] = f2[i][j] + if i > 0 { up_f2[i - 1][j] } else { 0 };
            up_f5[i][j] = f5[i][j] + if i > 0 { up_f5[i - 1][j] } else { 0 };
        }
    }

    let mut ans = 0;
    for i in 0..n {
        for j in 0..m {
            // 来到(i,j),计算6个方向上的因子数量之和
            let l2 = if j == 0 { 0 } else { left_f2[i][j - 1] }; // 左边的2因子数量之和 
            let l5 = if j == 0 { 0 } else { left_f5[i][j - 1] }; // 左边的5因子数量之和
            let r2 = left_f2[i][m - 1] - left_f2[i][j]; // 右边的2因子数量之和
            let r5 = left_f5[i][m - 1] - left_f5[i][j]; // 右边的5因子数量之和
            let u2 = if i == 0 { 0 } else { up_f2[i - 1][j] }; // 上面的2因子数量之和
            let u5 = if i == 0 { 0 } else { up_f5[i - 1][j] }; // 上面的5因子数量之和
            let d2 = up_f2[n - 1][j] - up_f2[i][j]; // 下面的2因子数量之和
            let d5 = up_f5[n - 1][j] - up_f5[i][j]; // 下面的5因子数量之和

            // 计算6个方向上因子数量之和最小的值
            let p1 = std::cmp::min(l2 + u2 + f2[i][j], l5 + u5 + f5[i][j]);
            let p2 = std::cmp::min(l2 + r2 + f2[i][j], l5 + r5 + f5[i][j]);
            let p3 = std::cmp::min(l2 + d2 + f2[i][j], l5 + d5 + f5[i][j]);
            let p4 = std::cmp::min(u2 + r2 + f2[i][j], u5 + r5 + f5[i][j]);
            let p5 = std::cmp::min(u2 + d2 + f2[i][j], u5 + d5 + f5[i][j]);
            let p6 = std::cmp::min(r2 + d2 + f2[i][j], r5 + d5 + f5[i][j]);

            // 取6个方向上的最小值中的最大值作为答案
            ans = std::cmp::max(
                ans,
                std::cmp::max(
                    std::cmp::max(p1, p2),
                    std::cmp::max(std::cmp::max(p3, p4), std::cmp::max(p5, p6)),
                ),
            );
        }
    }

    ans
}

// 计算num有几个f的因子
fn factors(num: i32, f: i32) -> i32 {
    let mut ans = 0;
    let mut num = num;

    while num % f == 0 {
        ans += 1;
        num /= f;
    }

    ans
}

fn main() {
    let matrix1 = vec![
        vec![5, 8, 3, 1],
        vec![4, 15, 12, 1],
        vec![6, 7, 10, 1],
        vec![9, 1, 2, 1],
    ];
    println!("{}", most_trailing_zeros(&matrix1)); // 输出:2

    let matrix2 = vec![
        vec![7500, 10, 11, 12],
        vec![6250, 13, 14, 15],
        vec![134, 17, 16, 1],
        vec![5500, 2093, 5120, 238],
    ];
    println!("{}", most_trailing_zeros(&matrix2)); // 输出:13
}

运行结果

在这里插入图片描述

本作品采用《CC 协议》,转载必须注明作者和本文链接
微信公众号:福大大架构师每日一题。最新面试题,涉及golang,rust,mysql,redis,云原生,算法,分布式,网络,操作系统。
讨论数量: 0
(= ̄ω ̄=)··· 暂无内容!

讨论应以学习和精进为目的。请勿发布不友善或者负能量的内容,与人为善,比聪明更重要!
未填写
文章
469
粉丝
21
喜欢
37
收藏
22
排名:457
访问:1.9 万
私信
所有博文
社区赞助商