LeetCode in Kotlin

788. Rotated Digits

Medium

An integer x is a good if after rotating each digit individually by 180 degrees, we get a valid number that is different from x. Each digit must be rotated - we cannot choose to leave it alone.

A number is valid if each digit remains a digit after rotation. For example:

Given an integer n, return the number of good integers in the range [1, n].

Example 1:

Input: n = 10

Output: 4

Explanation: There are four good numbers in the range [1, 10] : 2, 5, 6, 9. Note that 1 and 10 are not good numbers, since they remain unchanged after rotating.

Example 2:

Input: n = 1

Output: 0

Example 3:

Input: n = 2

Output: 1

Constraints:

Solution

class Solution {
    fun rotatedDigits(n: Int): Int {
        val flag = IntArray(n + 1)
        flag[0] = 2
        val indexesValueTwo = intArrayOf(1, 8)
        for (value in indexesValueTwo) {
            if (n >= value) {
                flag[value] = 2
            }
        }
        val indexesValueOne = intArrayOf(2, 5, 6, 9)
        for (value in indexesValueOne) {
            if (n >= value) {
                flag[value] = 1
            }
        }
        var rs = 0
        for (i in 1..n) {
            val residual = i % 10
            if (flag[residual] != 0) {
                if ((residual == 1 || residual == 0 || residual == 8) && flag[i / 10] == 2) {
                    flag[i] = 2
                    continue
                }
                if (flag[i / 10] != 0) {
                    flag[i] = 1
                    rs++
                }
            }
        }
        return rs
    }
}