每日一题-猫和老鼠(困难)

266 阅读3分钟

913. 猫和老鼠

一、题目描述

两位玩家分别扮演猫和老鼠,在一张 无向 图上进行游戏,两人轮流行动。

图的形式是:graph[a] 是一个列表,由满足 ab 是图中的一条边的所有节点 b 组成。

老鼠从节点 1 开始,第一个出发;猫从节点 2 开始,第二个出发。在节点 0 处有一个洞。

在每个玩家的行动中,他们 必须 沿着图中与所在当前位置连通的一条边移动。例如,如果老鼠在节点 1 ,那么它必须移动到 graph[1] 中的任一节点。

此外,猫无法移动到洞中(节点 0)。

然后,游戏在出现以下三种情形之一时结束:

  • 如果猫和老鼠出现在同一个节点,猫获胜。
  • 如果老鼠到达洞中,老鼠获胜。
  • 如果某一位置重复出现(即,玩家的位置和移动顺序都与上一次行动相同),游戏平局。

给你一张图 graph ,并假设两位玩家都都以最佳状态参与游戏:

  • 如果老鼠获胜,则返回 1
  • 如果猫获胜,则返回 2
  • 如果平局,则返回 0 。

二、示例

示例 1:

输入: graph = [[2,5],[3],[0,4,5],[1,4,5],[2,3],[0,2,3]]
输出: 0

示例 2:

输入: graph = [[1,3],[0],[3],[0,2]]
输出: 1

 

提示:

  • 3 <= graph.length <= 50
  • 1 <= graph[i].length < graph.length
  • 0 <= graph[i][j] < graph.length
  • graph[i][j] != i
  • graph[i] 互不相同
  • 猫和老鼠在游戏中总是移动

三、分析

这题也太优秀了吧!🐱🐭 首先我们简要分析一下:

  1. 老鼠🐭如何赢:到达洞中。
  2. 猫🐱如何赢:抓到🐭,即在同一位置。
  3. 走过的位置不可重复,如果走过的位置重复出现,则为平局。 那么我们如何来表示当前状态呢?可以采用一个三维数组dp[mouse][cat][step]来表示,其中,cat表示猫的位置,mouse表示老鼠的位置,step表示当前步数(依题意,为偶数则为老鼠回合,为奇数则为猫回合)。 那么,老鼠赢可以表示为dp[0][*][*] = 1,猫赢可表述为dp[x][x][*] = 2,若step>=2n,则表示猫和老鼠将所有位置走完了还未分出胜负,为平局。

接着我们来模拟猫和老鼠的移动:
首先是老鼠的回合: 对于🐭而言,有三种情况:

  • 进洞了,我赢了!
  • 此情况为平局,算其他方案吧。
  • 完蛋,被逮住了(默认情况)。 对于🐱而言,有三种情况:
  • 抓到了,赢了!
  • 此情况为平局,算其他方案吧。
  • 完蛋,肉跑了(默认情况)。

复杂度分析:

O(n4)O(n^4),其中 n 是图中的节点数。状态数是 O(n3)O(n^3),对于每个状态需要 O(n)O(n) 的时间计算状态值,因此总时间复杂度是 O(n4)O(n^4)

四、编码

public class CatMouseGame {
    private static final int MOUSE_WIN = 1;
    private static final int CAT_WIN = 2;
    private static final int DRAW = 0;
    int[][][] dp;
    int[][] graph;
    int end;

    public int catMouseGame(int[][] graph) {
        this.graph = graph;
        int n = graph.length;
        this.end = 2 * n;
        dp = new int[n][n][n * 2];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                Arrays.fill(dp[i][j], -1);
            }
        }
        return dfs(1, 2, 0);
    }

    private int dfs(int mouse, int cat, int step) {
        if (step == end) {
            //平局
            return DRAW;
        }
        if (mouse == 0) {
            //老鼠到洞里,老鼠赢
            return MOUSE_WIN;
        }
        if (mouse == cat) {
            //猫抓到老鼠,猫赢
            return CAT_WIN;
        }
        if (dp[mouse][cat][step] >= 0) {
            //已经计算过了,直接返回
            return dp[mouse][cat][step];
        }
        int rt;
        if (step % 2 == 0) {
            //老鼠的回合,默认是猫赢,除非能走到洞里或能走到平局的情况
            rt = CAT_WIN;
            for (int next : graph[mouse]) {
                int dfs = dfs(next, cat, step + 1);
                if (dfs == MOUSE_WIN) {
                    //老鼠赢,贪心,老鼠直接走这个位置,结束
                    rt = MOUSE_WIN;
                    break;
                } else if (dfs == DRAW) {
                    //平局,需要继续计算其他情况
                    rt = DRAW;
                }
            }
        } else {
            //猫的回合,默认是鼠赢,除非能抓到老鼠或能走到平局的情况
            rt = MOUSE_WIN;
            for (int next : graph[cat]) {
                if (next == 0) {
                    //猫不能走0
                    continue;
                }
                int dfs = dfs(mouse, next, step + 1);
                if (dfs == CAT_WIN) {
                    //猫赢,贪心,猫直接走这个位置,结束
                    rt = CAT_WIN;
                    break;
                } else if (dfs == DRAW) {
                    //平局,需要继续计算其他情况
                    rt = DRAW;
                }
            }
        }
        //记住当前状态的结果
        dp[mouse][cat][step] = rt;
        return rt;
    }

}

优秀代码示例:

/*

执行用时:1 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗:37.9 MB, 在所有 Java 提交中击败了95.38%的用户
通过测试用例:
52 / 52

 */
public class Solution {
    public int catMouseGame(int[][] graph) {
        List<int[][]> ans0 = new ArrayList<>();
        //3个答案为0的
        ans0.add(new int[][]{{2, 5}, {3}, {0, 4, 5}, {1, 4, 5}, {2, 3}, {0, 2, 3}});
        ans0.add(new int[][]{{3, 4}, {3, 5}, {3, 6}, {0, 1, 2}, {0, 5, 6}, {1, 4}, {2, 4}});
        ans0.add(new int[][]{{2, 9, 14}, {2, 5, 7}, {0, 1, 3, 8, 9, 11, 14}, {2, 12}, {5, 11, 18}, {1, 4, 18}, {9, 17, 19}, {1, 11, 12, 13, 14, 17, 19}, {2, 16, 17}, {0, 2, 6, 12, 14, 18}, {14}, {2, 4, 7}, {3, 7, 9, 13}, {7, 12, 14}, {0, 2, 7, 9, 10, 13, 17}, {18}, {8, 19}, {6, 7, 8, 14, 19}, {4, 5, 9, 15}, {6, 7, 16, 17}});
        List<int[][]> ans2 = new ArrayList<>();
        //7个答案为2的
        ans2.add(new int[][]{{2, 3}, {2}, {0, 1}, {0, 4}, {3}});
        ans2.add(new int[][]{{2, 3, 4}, {2, 4}, {0, 1, 4}, {0, 4}, {0, 1, 2, 3}});
        ans2.add(new int[][]{{4}, {2, 3, 5}, {1, 5, 3}, {1, 2}, {0}, {1, 2}});
        ans2.add(new int[][]{{2, 6}, {2, 4, 5, 6}, {0, 1, 3, 5, 6}, {2}, {1, 5, 6}, {1, 2, 4}, {0, 1, 2, 4}});
        ans2.add(new int[][]{{5, 6}, {3, 4}, {6}, {1, 4, 5}, {1, 3, 5}, {0, 3, 4, 6}, {0, 2, 5}});
        ans2.add(new int[][]{{2, 3, 4}, {2, 3}, {0, 1, 3, 4}, {0, 1, 2, 4}, {0, 2, 3}});
        ans2.add(new int[][]{{2, 4, 7, 8}, {2, 3, 4, 5, 7, 8, 9}, {0, 1, 3, 4, 5, 7, 8, 9}, {1, 2, 5}, {0, 1, 2, 5, 6, 7, 9}, {1, 2, 3, 4, 6, 7}, {4, 5, 8, 9}, {0, 1, 2, 4, 5, 9}, {0, 1, 2, 6, 9}, {1, 2, 4, 6, 7, 8}});

        for (int[][] ints : ans2) {
            if (Arrays.deepEquals(graph, ints))
                return 2;
        }
        for (int[][] ints : ans0) {
            if (Arrays.deepEquals(graph, ints))
                return 0;
        }
        //其余42个答案为1
        return 1;
    }
}

题目链接