游戏队友搜索-题目解析| 豆包MarsCode AI刷题

235 阅读7分钟

题目解析

问题描述

在一款多人游戏中,每局比赛需要多个玩家参与。如果发现两名玩家至少一起玩过两局比赛,则可以认为这两名玩家互为队友。现在你有一份玩家(通过玩家ID标识)和比赛局次(通过比赛ID标识)的历史记录表,目标是帮助某位指定玩家找到所有符合条件的队友。

输入

id:指定玩家的ID.

num:历史记录的数量。

array:一个二维数组,表示玩家和比赛的历史记录,每个元素是一个长度为2的数组,第一个元素是玩家D,第二个元素是比赛ID。

输出

一个数组,包含所有与指定玩家共同参与至少两局比赛的玩家D,按升序排列。

思路

一、数据结构选择

使用HashMap<Integer,Set来存储每个玩家参与的游戏列表,其中键是玩家ID,值是该玩家参与的游戏ID

二、算法步骤

1、遍历输入的记录,填充playerGames映射。

2、获取指定玩家参与的游戏列表targetGames.

3、如果指定玩家没有参与任何游戏,返回空数组。

4、遍历所有玩家,计算每个玩家与指定玩家共同参与的游戏数量。

5、如果共同参与的游戏数量至少为2,则将该玩家ID加入teammates集合。

6、将teammates集合转换为数组并排序,返回结果。

图解

假设我们有以下输入:

    id = 1
num = 10
array = [[1, 1], [1, 2], [1, 3], [2, 1], [2, 4], [3, 2], [4, 1], [4, 2], [5, 2], [5, 3]]  

填充playerGames映射

    playerGames = {
    1: {1, 2, 3},
    2: {1, 4},
    3: {2},
    4: {1, 2},
    5: {2, 3}
}

获取指定玩家参与的游戏列表

    targetGames = {1, 2, 3}

遍历所有玩家,计算共同参与的游戏数量

1、玩家2:共同游戏数量=1(游戏1)

2、玩家3:共同游戏数量=1(游戏2)

3、玩家4:共同游戏数量=2(游戏1,2)

4、玩家5:共同游戏数量=2(游戏2,3)

找出队友

玩家4和玩家5是队友,因为他们与玩家1共同参与了至少两局游戏。

返回结果

    result = [4, 5]

代码详解

    import java.util.*;

public class Main {
    public static void main(String[] args) {
        // Add your test cases here
        System.out.println(Arrays.equals(solution(1, 10, new int[][]{{1, 1}, {1, 2}, {1, 3}, {2, 1}, {2, 4}, {3, 2}, {4, 1}, {4, 2}, {5, 2}, {5, 3}}), new int[]{4, 5}));
    }

    public static int[] solution(int id, int num, int[][] array) {
        // 使用HashMap来存储每个玩家参与的游戏列表
        Map<Integer, Set<Integer>> playerGames = new HashMap<>();

        // 遍历所有记录,填充playerGames
        for (int[] record : array) {
            int playerId = record[0];
            int gameId = record[1];
            playerGames.putIfAbsent(playerId, new HashSet<>());
            playerGames.get(playerId).add(gameId);
        }

        // 获取指定玩家参与的游戏列表
        Set<Integer> targetGames = playerGames.get(id);

        // 如果指定玩家没有参与任何游戏,返回空数组
        if (targetGames == null) {
            return new int[0];
        }

        // 使用HashSet来存储队友ID
        Set<Integer> teammates = new HashSet<>();

        // 遍历所有玩家,找出与指定玩家共同参与至少两局游戏的玩家
        for (Map.Entry<Integer, Set<Integer>> entry : playerGames.entrySet()) {
            int playerId = entry.getKey();
            if (playerId == id) continue; // 跳过指定玩家自己

            Set<Integer> games = entry.getValue();
            int commonGamesCount = 0;

            // 计算共同参与的游戏数量
            for (int game : games) {
                if (targetGames.contains(game)) {
                    commonGamesCount++;
                }
            }

            // 如果共同参与的游戏数量至少为2,则认为是队友
            if (commonGamesCount >= 2) {
                teammates.add(playerId);
            }
        }

        // 将队友ID转换为数组并排序
        int[] result = new int[teammates.size()];
        int index = 0;
        for (int teammate : teammates) {
            result[index++] = teammate;
        }
        Arrays.sort(result);

        return result;
    }
}

代码解释

数据结构选择:

Map<Integer,SetplayerGames:用于存储每个玩家参与的游戏列表。

填充playerGames

遍历输入的记录,将每个玩家参与的游戏心添加到对应的集合中。

获取指定玩家的游戏列表

·从playerGames 中获取指定玩家的游戏列表targetGames.

检查targetGames是否为nuLL

·如果targetGames为nuLL,说明指定玩家没有参与任何游戏,返回空数组。

遍历所有玩家,计算共同参与的游戏数量

遍历playerGames中的每个玩家,计算与指定玩家共同参与的游戏数量。 如果共同参与的游戏数量至少为2,则将该玩家ID加入teammates集合。

返回结果

将teammates集合转换为数组并排序,返回结果。

优化建议

1.减少重复计算:使用Set的交集操作来减少重复计算,例如使用Set.retainAll方法。

2.提前退出:在计算共同参与的游戏数量时,如果已经达到2个共同游戏,提前退出内层循环,减少不必要的计算。

3.并行处理:如果玩家数量非常多,考虑使用并行处理来加速计算。

总结

通过上述详细的解析和代码实现,我们不仅解决了特定的问题——找到与指定玩家共同参与至少两局比赛的队友,还深入探讨了问题的解决方法、数据结构的选择、算法的优化等方面。这一过程不仅加深了对编程和算法的理解,也为解决类似问题提供了宝贵的经验和思路。 在面对此类问题时,我们首先明确问题的要求,然后选择合适的数据结构来存储和处理数据。在本题中,我们使用HashMap<Integer, Set<Integer>>来存储每个玩家参与的游戏列表,这种数据结构的选择使得我们能够高效地查询和计算玩家之间的共同游戏数量。 接着,我们按照清晰的步骤来解决问题:填充映射、获取指定玩家的游戏列表、遍历所有玩家计算共同游戏数量、找出队友并返回结果。每一步都紧密相连,逻辑清晰,确保了问题的正确解决。 数据结构和算法的选择对于问题的解决至关重要。在本题中,HashMapSet的结合使用使得我们能够高效地存储和查询数据。同时,我们采用了简单的嵌套循环来计算共同游戏数量,虽然这种方法在数据量较大时可能效率不高,但对于本题来说已经足够应对。 在实际应用中,我们还需要根据问题的具体要求和数据的规模来选择合适的数据结构和算法。例如,在数据量非常大时,可以考虑使用更高效的数据结构(如位图)或算法(如并行计算)来提高性能。 在解决问题的过程中,我们不断思考如何优化算法以提高性能。例如,使用Set的交集操作来减少重复计算,或者在计算共同游戏数量时提前退出内层循环以减少不必要的计算。这些优化措施虽然看似微小,但在处理大规模数据时能够显著提高性能。 此外,我们还可以考虑使用更高级的算法或数据结构来进一步优化解决方案。例如,可以使用图论中的算法来处理玩家之间的游戏关系,或者使用数据库索引来加速查询操作。 在编写代码时,我们注重代码的可读性和可维护性。通过清晰的变量命名、合理的代码结构和详细的注释,使得代码易于理解和维护。这不仅有助于我们自己的代码复用和调试,也方便了其他开发者对代码的阅读和修改。 本题虽然是一个具体的编程问题,但其背后的思想和方法却具有广泛的应用价值。例如,在社交网络分析中,我们可以使用类似的方法来找出用户之间的共同兴趣或活动;在推荐系统中,我们可以根据用户之间的共同行为来推荐相似的内容或产品。 此外,我们还可以将本题的方法拓展到更复杂的场景中。例如,考虑玩家之间的游戏时间、游戏类型等因素,来更精确地找出符合条件的队友;或者将问题扩展到多人团队中,找出共同参与多局比赛的团队成员。 综上所述,通过本题我们不仅解决了一个具体的编程问题,还深入探讨了解决问题的策略、数据结构与算法的选择、优化的重要性以及代码的可读性和可维护性等方面。这些经验和思路将对我们未来的学习和工作产生积极的影响。