g0601_0700.s0649_dota2_senate.Solution.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of leetcode-in-kotlin Show documentation

Kotlin-based LeetCode algorithm problem solutions, regularly updated

There is a newer version: 1.30

package g0601_0700.s0649_dota2_senate

// #Medium #String #Greedy #Queue #2023_02_12_Time_217_ms_(100.00%)_Space_36.1_MB_(100.00%)

class Solution {
    fun predictPartyVictory(senate: String): String {
        val blocks = IntArray(2)
        val status = BooleanArray(senate.length)
        var changes = true
        while (changes) {
            changes = false
            for (i in senate.indices) {
                if (status[i]) {
                    continue
                }
                val curr = senate[i]
                val block: Int = if (curr == 'R') {
                    0
                } else {
                    1
                }
                if (blocks[1 - block] > 0) {
                    status[i] = true
                    blocks[1 - block]--
                    changes = true
                } else {
                    blocks[block]++
                }
            }
        }
        for (i in senate.indices) {
            if (!status[i]) {
                return if (senate[i] == 'R') {
                    "Radiant"
                } else {
                    "Dire"
                }
            }
        }
        return ""
    }
}