LeetCode in Kotlin

659. Split Array into Consecutive Subsequences

Medium

You are given an integer array nums that is sorted in non-decreasing order.

Determine if it is possible to split nums into one or more subsequences such that both of the following conditions are true:

Return true if you can split nums according to the above conditions, or false otherwise.

A subsequence of an array is a new array that is formed from the original array by deleting some (can be none) of the elements without disturbing the relative positions of the remaining elements. (i.e., [1,3,5] is a subsequence of [1,2,3,4,5] while [1,3,2] is not).

Example 1:

Input: nums = [1,2,3,3,4,5]

Output: true

Explanation: nums can be split into the following subsequences:

[1,2,3,3,4,5] –> 1, 2, 3

[1,2,3,**<ins>3</ins>**,**<ins>4</ins>**,**<ins>5</ins>**] --> 3, 4, 5

Example 2:

Input: nums = [1,2,3,3,4,4,5,5]

Output: true

Explanation: nums can be split into the following subsequences:

[1,2,3,3,4,4,5,5] –> 1, 2, 3, 4, 5

[1,2,3,3,4,4,5,5] –> 3, 4, 5

Example 3:

Input: nums = [1,2,3,4,4,5]

Output: false

Explanation: It is impossible to split nums into consecutive increasing subsequences of length 3 or more.

Constraints:

Solution

class Solution {
    fun isPossible(nums: IntArray): Boolean {
        val element = IntArray(2001)
        for (num in nums) {
            element[num + 1000] += 1
        }
        for (i in element.indices) {
            while (element[i] > 0) {
                var length = 1
                while (i + length < element.size &&
                    element[i + length] >= element[i + length - 1]
                ) {
                    length += 1
                }
                if (length < 3) {
                    return false
                } else {
                    for (j in i until i + length) {
                        element[j] -= 1
                    }
                }
            }
        }
        return true
    }
}