LeetCode in Kotlin

2197. Replace Non-Coprime Numbers in Array

Hard

You are given an array of integers nums. Perform the following steps:

  1. Find any two adjacent numbers in nums that are non-coprime.
  2. If no such numbers are found, stop the process.
  3. Otherwise, delete the two numbers and replace them with their LCM (Least Common Multiple).
  4. Repeat this process as long as you keep finding two adjacent non-coprime numbers.

Return the final modified array. It can be shown that replacing adjacent non-coprime numbers in any arbitrary order will lead to the same result.

The test cases are generated such that the values in the final array are less than or equal to 108.

Two values x and y are non-coprime if GCD(x, y) > 1 where GCD(x, y) is the Greatest Common Divisor of x and y.

Example 1:

Input: nums = [6,4,3,2,7,6,2]

Output: [12,7,6]

Explanation:

There are no more adjacent non-coprime numbers in nums.

Thus, the final modified array is [12,7,6].

Note that there are other ways to obtain the same resultant array.

Example 2:

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

Output: [2,1,1,3]

Explanation:

There are no more adjacent non-coprime numbers in nums.

Thus, the final modified array is [2,1,1,3].

Note that there are other ways to obtain the same resultant array.

Constraints:

Solution

class Solution {
    fun replaceNonCoprimes(nums: IntArray): List<Int> {
        val res = ArrayList<Int>()
        var i = 1
        res.add(nums[0])
        while (i < nums.size) {
            val first = res[res.size - 1]
            val second = nums[i]
            val gcd = gcd(first, second)
            if (gcd > 1) {
                val lcm = first.toLong() * second.toLong() / gcd
                if (res.isNotEmpty()) {
                    res.removeAt(res.size - 1)
                }
                res.add(lcm.toInt())
                recursivelyCheck(res)
            } else {
                res.add(second)
            }
            i++
        }
        return res
    }

    private fun gcd(a: Int, b: Int): Int {
        if (a > b) {
            return gcd(b, a)
        }
        return if (b % a == 0) {
            a
        } else {
            gcd(b % a, a)
        }
    }

    private fun recursivelyCheck(list: ArrayList<Int>) {
        if (list.size < 2) {
            return
        }
        val a = list.removeAt(list.size - 1)
        val b = list.removeAt(list.size - 1)
        val gcd = gcd(a, b)
        if (gcd > 1) {
            val lcm = a.toLong() * b.toLong() / gcd
            list.add(lcm.toInt())
            recursivelyCheck(list)
        } else {
            list.add(b)
            list.add(a)
        }
    }
}