Question

2639. 查询网格图中每一列的宽度

English Version

题目描述

给你一个下标从 0 开始的 m x n 整数矩阵 grid 。矩阵中某一列的宽度是这一列数字的最大 字符串长度 。

• 比方说，如果 grid = [[-10], [3], [12]] ，那么唯一一列的宽度是 3 ，因为 -10 的字符串长度为 3 。

请你返回一个大小为 n 的整数数组 ans ，其中 ans[i] 是第 i 列的宽度。

一个有 len 个数位的整数 x ，如果是非负数，那么 字符串长度 为 len ，否则为 len + 1 。

 

示例 1：

输入：grid = [[1],[22],[333]]
输出：[3]
解释：第 0 列中，333 字符串长度为 3 。

示例 2：

输入：grid = [[-15,1,3],[15,7,12],[5,6,-2]]
输出：[3,1,2]
解释：
第 0 列中，只有 -15 字符串长度为 3 。
第 1 列中，所有整数的字符串长度都是 1 。
第 2 列中，12 和 -2 的字符串长度都为 2 。

 

提示：

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 100
• -109 <= grid[r][c] <= 109

解法

方法一：模拟

我们记矩阵的列数为 $n$，创建一个长度为 $n$ 的数组 $ans$，其中 $ans[i]$ 表示第 $i$ 列的宽度。初始时，$ans[i]$ 的值均为 $0$。

遍历矩阵中的每一行，对于每一行中的每个元素，计算其字符串长度 $w$，并更新 $ans[j]$ 的值为 $max(ans[j], w)$。

遍历完所有行后，数组 $ans$ 中的每个元素即为对应列的宽度。

时间复杂度 $O(m \times n)$，空间复杂度 $O(\log M)$。其中 $m$ 和 $n$ 分别为矩阵的行数和列数，而 $M$ 为矩阵中的最大元素绝对值。

class Solution:
  def findColumnWidth(self, grid: List[List[int]]) -> List[int]:
    ans = [0] * len(grid[0])
    for row in grid:
      for j, x in enumerate(row):
        w = len(str(x))
        ans[j] = max(ans[j], w)
    return ans

class Solution {
  public int[] findColumnWidth(int[][] grid) {
    int n = grid[0].length;
    int[] ans = new int[n];
    for (var row : grid) {
      for (int j = 0; j < n; ++j) {
        int w = String.valueOf(row[j]).length();
        ans[j] = Math.max(ans[j], w);
      }
    }
    return ans;
  }
}

class Solution {
public:
  vector<int> findColumnWidth(vector<vector<int>>& grid) {
    int n = grid[0].size();
    vector<int> ans(n);
    for (auto& row : grid) {
      for (int j = 0; j < n; ++j) {
        int w = to_string(row[j]).size();
        ans[j] = max(ans[j], w);
      }
    }
    return ans;
  }
};

func findColumnWidth(grid [][]int) []int {
  ans := make([]int, len(grid[0]))
  for _, row := range grid {
    for j, x := range row {
      w := len(strconv.Itoa(x))
      ans[j] = max(ans[j], w)
    }
  }
  return ans
}

function findColumnWidth(grid: number[][]): number[] {
  const n = grid[0].length;
  const ans: number[] = new Array(n).fill(0);
  for (const row of grid) {
    for (let j = 0; j < n; ++j) {
      const w: number = String(row[j]).length;
      ans[j] = Math.max(ans[j], w);
    }
  }
  return ans;
}

impl Solution {
  pub fn find_column_width(grid: Vec<Vec<i32>>) -> Vec<i32> {
    let mut ans = vec![0; grid[0].len()];

    for row in grid.iter() {
      for (j, num) in row.iter().enumerate() {
        let width = num.to_string().len() as i32;
        ans[j] = std::cmp::max(ans[j], width);
      }
    }

    ans
  }
}

方法二

class Solution:
  def findColumnWidth(self, grid: List[List[int]]) -> List[int]:
    return [max(len(str(x)) for x in col) for col in zip(*grid)]