913.猫和老鼠
最编程
2024-05-04 09:35:45
...
两位玩家分别扮演猫和老鼠,在一张 无向 图上进行游戏,两人轮流行动。
图的形式是:graph[a] 是一个列表,由满足 ab 是图中的一条边的所有节点 b 组成。
老鼠从节点 1 开始,第一个出发;猫从节点 2 开始,第二个出发。在节点 0 处有一个洞。
在每个玩家的行动中,他们 必须 沿着图中与所在当前位置连通的一条边移动。例如,如果老鼠在节点 1 ,那么它必须移动到 graph[1] 中的任一节点。
此外,猫无法移动到洞中(节点 0)。
然后,游戏在出现以下三种情形之一时结束:
如果猫和老鼠出现在同一个节点,猫获胜。
如果老鼠到达洞中,老鼠获胜。
如果某一位置重复出现(即,玩家的位置和移动顺序都与上一次行动相同),游戏平局。
给你一张图 graph ,并假设两位玩家都都以最佳状态参与游戏:
如果老鼠获胜,则返回 1;
如果猫获胜,则返回 2;
如果平局,则返回 0 。
import java.util.Arrays;
import java.util.Scanner;
class Solution {
private final int MOUSE_WIN = 1;
private final int CAT_WIN = 2;
private final int NO_WIN = 0;
private int n;
private int[][][] dp;
private int[][] graph;
public int catMouseGame(int[][] graph) {
this.n = graph.length;
this.graph = graph;
this.dp = new int[graph.length][graph.length][graph.length * 2];
for (int i = 0; i < dp.length; ++i) {
for (int j = 0; j < dp[0].length; ++j) {
Arrays.fill(dp[i][j], -1);
}
}
return solve(1, 2, 0);
}
private int solve(int mouse, int cat, int turns) {
if (turns == 2 * n) {
return NO_WIN;
}
if (dp[mouse][cat][turns] == -1) {
if (mouse == 0) {
dp[mouse][cat][turns] = MOUSE_WIN;
} else if (mouse == cat) {
dp[mouse][cat][turns] = CAT_WIN;
} else {
dp[mouse][cat][turns] = whoWin(mouse, cat, turns);
}
}
return dp[mouse][cat][turns];
}
private int whoWin(int mouse, int cat, int turns) {
boolean isMouse = turns % 2 == 0;
int from = isMouse ? mouse : cat;
int defaultRes = isMouse ? CAT_WIN : MOUSE_WIN;
int ans = defaultRes;
for (int to : graph[from]) {
if (!isMouse && to == 0) {
continue;
}
int nextMouse = isMouse ? to : mouse;
int nextCat = isMouse ? cat : to;
int res = solve(nextMouse, nextCat, turns + 1);
if (res != defaultRes) {
ans = res;
if (res != NO_WIN) {
break;
}
}
}
return ans;
}
}
上一篇: 猫鼠迷宫问题
下一篇: 猫叫老鼠跑 C# 委托与事件