LeetCode in Kotlin

1545. Find Kth Bit in Nth Binary String

Medium

Given two positive integers n and k, the binary string Sn is formed as follows:

• S1 = "0"
• Si = Si - 1 + "1" + reverse(invert(Si - 1)) for i > 1

Where + denotes the concatenation operation, reverse(x) returns the reversed string x, and invert(x) inverts all the bits in x (0 changes to 1 and 1 changes to 0).

For example, the first four strings in the above sequence are:

• S1 = "0"
• S2 = "0**1**1"
• S3 = "011**1**001"
• S4 = "0111001**1**0110001"

Return the kth bit in Sn. It is guaranteed that k is valid for the given n.

Example 1:

Input: n = 3, k = 1

Output: “0”

Explanation: S₃ is “0111001”. The 1^st bit is “0”.

Example 2:

Input: n = 4, k = 11

Output: “1”

Explanation: S₄ is “011100110110001”. The 11^th bit is “1”.

Constraints:

• 1 <= n <= 20
• 1 <= k <= 2n - 1

Solution

@Suppress("NAME_SHADOWING", "UNUSED_PARAMETER")
class Solution {
    fun findKthBit(n: Int, k: Int): Char {
        var k = k
        var flip = false
        while (k != 1) {
            val base = floorTwo(k)
            if (base == k) {
                return if (flip) '0' else '1'
            }
            flip = !flip
            k = base - (k - base)
        }
        return if (flip) '1' else '0'
    }

    private fun floorTwo(k: Int): Int {
        var k = k
        while (k and k - 1 > 0) {
            k = k and k - 1
        }
        return k
    }
}

This site is open source. Improve this page.