LeetCode in Kotlin

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

class Solution {
    fun findColumnWidth(grid: Array<IntArray>): IntArray {
        val ans = IntArray(grid[0].size) { 1 }
        for (j in 0 until grid[0].size) {
            var max = 1
            for (i in 0 until grid.size) {
                var num = grid[i][j]
                var neg = false
                if (num < 0) {
                    neg = true
                    num *= -1
                }
                var size = 0
                while (num > 0) {
                    num /= 10
                    size++
                }
                if (neg) {
                    size += 1
                }
                max = maxOf(max, size)
            }
            ans[j] = max
        }
        return ans
    }
}