LeetCode-in-Java

2639. Find the Width of Columns of a Grid

Easy

You are given a 0-indexed m x n integer matrix grid. The width of a column is the maximum length of its integers.

Return an integer array ans of size n where ans[i] is the width of the ith column.

The length of an integer x with len digits is equal to len if x is non-negative, and len + 1 otherwise.

Example 1:

Input: grid = [[1],[22],[333]]

Output: [3]

Explanation: In the 0th column, 333 is of length 3.

Example 2:

Input: grid = [[-15,1,3],[15,7,12],[5,6,-2]]

Output: [3,1,2]

Explanation:

In the 0th column, only -15 is of length 3.

In the 1st column, all integers are of length 1.

In the 2nd column, both 12 and -2 are of length 2.

Constraints:

Solution

public class Solution {
    public int[] findColumnWidth(int[][] grid) {
        int[] ans = new int[grid[0].length];
        for (int j = 0; j < grid[0].length; j++) {
            int max = 1;
            for (int[] ints : grid) {
                int num = ints[j];
                boolean neg = false;
                if (num < 0) {
                    neg = true;
                    num *= -1;
                }
                int size = 0;
                while (num > 0) {
                    num /= 10;
                    size++;
                }
                if (neg) {
                    size += 1;
                }
                max = Math.max(max, size);
            }
            ans[j] = max;
        }
        return ans;
    }
}