除客皆刀

今天 leetcode 的每日一题有点意思，第一次遇到和热门（DEAD）游戏相关的题目，还是我大 doto 的，虽然看完题目感觉有点像是硬往 dota2 上靠的…

题目内容

Dota2 的世界里有两个阵营：Radiant(天辉)和 Dire(夜魇)

Dota2 参议院由来自两派的参议员组成。现在参议院希望对一个 Dota2 游戏里的改变作出决定。他们以一个基于轮为过程的投票进行。在每一轮中，每一位参议员都可以行使两项权利中的一项：

禁止一名参议员的权利：

参议员可以让另一位参议员在这一轮和随后的几轮中丧失所有的权利。

宣布胜利：

      如果参议员发现有权利投票的参议员都是同一个阵营的，他可以宣布胜利并决定在游戏中的有关变化。

给定一个字符串代表每个参议员的阵营。字母 “R” 和 “D” 分别代表了 Radiant（天辉）和 Dire（夜魇）。然后，如果有 n 个参议员，给定字符串的大小将是 n。

以轮为基础的过程从给定顺序的第一个参议员开始到最后一个参议员结束。这一过程将持续到投票结束。所有失去权利的参议员将在过程中被跳过。

假设每一位参议员都足够聪明，会为自己的政党做出最好的策略，你需要预测哪一方最终会宣布胜利并在 Dota2 游戏中决定改变。输出应该是 Radiant 或 Dire。

示例 1：

输入：“RD” 输出：“Radiant” 解释：第一个参议员来自 Radiant 阵营并且他可以使用第一项权利让第二个参议员失去权力，因此第二个参议员将被跳过因为他没有任何权利。然后在第二轮的时候，第一个参议员可以宣布胜利，因为他是唯一一个有投票权的人

示例 2：

输入：“RDD” 输出：“Dire” 解释： 第一轮中,第一个来自 Radiant 阵营的参议员可以使用第一项权利禁止第二个参议员的权利 第二个来自 Dire 阵营的参议员会被跳过因为他的权利被禁止 第三个来自 Dire 阵营的参议员可以使用他的第一项权利禁止第一个参议员的权利 因此在第二轮只剩下第三个参议员拥有投票的权利,于是他可以宣布胜利

解题思路

根据题干，很容易想出一个贪心的策略：每轮投票时，每个正在投票的参议员都行使权力 ban 掉后面最先投票的敌方参议员，直到有一方被 ban 光了，另一方就直接进行一个胜利的宣布。因此，可以使用两个队列保存双方阵营的参议员投票次序，开始时，遍历字符串，将每个参议员的投票次序放到相应队列，然后同时遍历双方队列。每次从双方队列中 poll 出第一个参议员的投票次序并进行比较，如果 radiant 的次序小，说明天辉先投票，按照贪心策略，天辉方肯定会 ban 掉夜宴方刚刚 poll 出来的那一个参议员，然后把自己重新放到队列末尾进行下一轮投票；如果 dire 的次序小，说明夜宴先投票，同样ban掉对方 poll 出来的参议员，将自己重新入队进行下一轮投票。重新入队时，投票次序=当前投票次序+给定的所有参议员的人数。遍历结束后，队列不为空的一方即可宣布胜利。

实现代码

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

fun predictPartyVictory(senate: String): String {
    val radiant = ArrayDeque<Int>()
    val dire = ArrayDeque<Int>()
    for (i in senate.indices) {
        if (senate[i] == 'R') {
            radiant.offer(i)
        } else {
            dire.offer(i)
        }
    }
    while (radiant.isNotEmpty() && dire.isNotEmpty()) {
        val r = radiant.poll()
        val d = dire.poll()
        if (r < d) {
            radiant.offer(r + senate.length)
        } else {
            dire.offer(d + senate.length)
        }
    }
    return if (radiant.isEmpty()) {
        "Dire"
    } else {
        "Radiant"
    }
}