LeetCode in Kotlin

650. 2 Keys Keyboard

Medium

There is only one character 'A' on the screen of a notepad. You can perform one of two operations on this notepad for each step:

Given an integer n, return the minimum number of operations to get the character 'A' exactly n times on the screen.

Example 1:

Input: n = 3

Output: 3

Explanation: Initially, we have one character ‘A’.

In step 1, we use Copy All operation.

In step 2, we use Paste operation to get ‘AA’.

In step 3, we use Paste operation to get ‘AAA’.

Example 2:

Input: n = 1

Output: 0

Constraints:

Solution

class Solution {
    fun minSteps(n: Int): Int {
        var count = 1
        var cost = 0
        var addValue = 1
        while (count < n) {
            cost++
            count += addValue
            if (n % count == 0) {
                cost++
                addValue = count
            }
        }
        return cost
    }
}