LeetCode in Kotlin

1195. Fizz Buzz Multithreaded

Medium

You have the four functions:

You are given an instance of the class FizzBuzz that has four functions: fizz, buzz, fizzbuzz and number. The same instance of FizzBuzz will be passed to four different threads:

Modify the given class to output the series [1, 2, "Fizz", 4, "Buzz", ...] where the ith token (1-indexed) of the series is:

Implement the FizzBuzz class:

Example 1:

Input: n = 15

Output: [1,2,”fizz”,4,”buzz”,”fizz”,7,8,”fizz”,”buzz”,11,”fizz”,13,14,”fizzbuzz”]

Example 2:

Input: n = 5

Output: [1,2,”fizz”,4,”buzz”]

Constraints:

Solution

import java.util.concurrent.atomic.AtomicInteger
import java.util.function.IntConsumer

class FizzBuzz(private val n: Int) {
    private val count = AtomicInteger(1)

    // printFizz.run() outputs "fizz".
    @Throws(InterruptedException::class)
    fun fizz(printFizz: Runnable) {
        var i: Int
        while (count.get().also { i = it } <= n) {
            if (i % 3 == 0 && i % 5 != 0) {
                printFizz.run()
                count.compareAndSet(i, i + 1)
            }
        }
    }

    // printBuzz.run() outputs "buzz".
    @Throws(InterruptedException::class)
    fun buzz(printBuzz: Runnable) {
        var i: Int
        while (count.get().also { i = it } <= n) {
            count.get()
            if (i % 5 == 0 && i % 3 != 0) {
                printBuzz.run()
                count.compareAndSet(i, i + 1)
            }
        }
    }

    // printFizzBuzz.run() outputs "fizzbuzz".
    @Throws(InterruptedException::class)
    fun fizzbuzz(printFizzBuzz: Runnable) {
        var i: Int
        while (count.get().also { i = it } <= n) {
            if (i % 15 == 0) {
                printFizzBuzz.run()
                count.compareAndSet(i, i + 1)
            }
        }
    }

    // printNumber.accept(x) outputs "x", where x is an integer.
    @Throws(InterruptedException::class)
    fun number(printNumber: IntConsumer) {
        var i: Int
        while (count.get().also { i = it } <= n) {
            if (i % 5 != 0 && i % 3 != 0) {
                printNumber.accept(i)
                count.compareAndSet(i, i + 1)
            }
        }
    }
}