当前位置：首页 > news >正文

LeetCode算法题(Go语言实现)_28

news 来源：原创 2025/6/29 21:04:56

题目

Dota2 的世界里有两个阵营：Radiant（天辉）和 Dire（夜魇）
Dota2 参议院由来自两派的参议员组成。现在参议院希望对一个 Dota2 游戏里的改变作出决定。他们以一个基于轮为过程的投票进行。在每一轮中，每一位参议员都可以行使两项权利中的一项：
禁止一名参议员的权利：参议员可以让另一位参议员在这一轮和随后的几轮中丧失所有的权利。
宣布胜利：如果参议员发现有权利投票的参议员都是同一个阵营的，他可以宣布胜利并决定在游戏中的有关变化。
给你一个字符串 senate 代表每个参议员的阵营。字母 ‘R’ 和 'D’分别代表了 Radiant（天辉）和 Dire（夜魇）。然后，如果有 n 个参议员，给定字符串的大小将是 n。
以轮为基础的过程从给定顺序的第一个参议员开始到最后一个参议员结束。这一过程将持续到投票结束。所有失去权利的参议员将在过程中被跳过。
假设每一位参议员都足够聪明，会为自己的政党做出最好的策略，你需要预测哪一方最终会宣布胜利并在 Dota2 游戏中决定改变。输出应该是 “Radiant” 或 “Dire” 。

一、代码实现

func predictPartyVictory(senate string) string {
    radiant := make([]int, 0)
    dire := make([]int, 0)
    n := len(senate)
    
    // 初始化队列，记录每个议员的位置
    for i, ch := range senate {
        if ch == 'R' {
            radiant = append(radiant, i)
        } else {
            dire = append(dire, i)
        }
    }
    
    // 模拟投票过程
    for len(radiant) > 0 && len(dire) > 0 {
        r := radiant[0]
        d := dire[0]
        radiant = radiant[1:]
        dire = dire[1:]
        
        // 位置靠前的议员禁止对方，并进入下一轮
        if r < d {
            radiant = append(radiant, r + n)
        } else {
            dire = append(dire, d + n)
        }
    }
    
    if len(radiant) > 0 {
        return "Radiant"
    }
    return "Dire"
}

二、算法分析

1. 核心思路

贪心策略：每个议员优先禁止下一个最近的敌方议员，以最大化己方优势
循环队列模拟：用两个队列分别存储双方议员的位置，通过位置比较实现投票轮次模拟

2. 关键步骤

初始化队列：遍历字符串，将R/D的位置分别存入两个队列
循环比较：每次取队列头部元素比较位置，较小者（先投票）禁止对方议员
下一轮入队：胜者将自己的位置加n（总人数）后重新入队，模拟循环投票
终止条件：当某一队列为空时，另一方获胜

3. 复杂度

指标	值	说明
时间复杂度	O(n)	每个元素最多入队、出队各一次
空间复杂度	O(n)	存储所有参议员位置

三、图解示例

在这里插入图片描述

四、边界条件与扩展

1. 特殊场景验证

全R/D字符串：直接返回对应阵营
交替序列：如"RDRDRD" → 最终由最后一个议员决定胜负
单议员场景：输入长度1时直接返回对应阵营

2. 多语言实现

# Python实现（deque优化）
from collections import deque

def predictPartyVictory(senate: str) -> str:
    radiant = deque()
    dire = deque()
    n = len(senate)
    for i, ch in enumerate(senate):
        if ch == 'R':
            radiant.append(i)
        else:
            dire.append(i)
    while radiant and dire:
        r = radiant.popleft()
        d = dire.popleft()
        if r < d:
            radiant.append(r + n)
        else:
            dire.append(d + n)
    return "Radiant" if radiant else "Dire"

// Java实现（LinkedList）
public String predictPartyVictory(String senate) {
    Queue<Integer> radiant = new LinkedList<>();
    Queue<Integer> dire = new LinkedList<>();
    int n = senate.length();
    for (int i = 0; i < n; i++) {
        if (senate.charAt(i) == 'R') radiant.offer(i);
        else dire.offer(i);
    }
    while (!radiant.isEmpty() && !dire.isEmpty()) {
        int r = radiant.poll(), d = dire.poll();
        if (r < d) radiant.offer(r + n);
        else dire.offer(d + n);
    }
    return radiant.isEmpty() ? "Dire" : "Radiant";
}