题意:
思路:
采用记忆化搜索,dp[t][x][y]表示走了t步后老鼠在x猫在y时的状态。
初始将dp数组都设为-1,表示未被经过。
dfs搜索,传的参数未当前的步数t,老鼠的位置x,猫的位置y。然后进行判断:
如果当前的步数>=2n 返回0平局
如果x=y 则猫赢 返回2
如果x=0 老鼠赢 返回1
对老鼠而言,遍历相邻的节点时 如果返回值为1 则老鼠肯定赢;如果返回值为2,则老鼠可以不走这一步;如果返回值为0,则可能是平局,但猫一定赢不了,所以标记flag=false;最后判断flag的值,如果flag为true的话,说明是老鼠输了;否则,为平局。
猫的过程也同理。
注意:猫不能走0
代码:
class Solution { public: int catMouseGame(vector<vector<int>>& graph) { int n=graph.size(); vector<vector<vector<int>>>dp(2*n,vector<vector<int>>(n,vector<int>(n,-1))); return dfs(graph,0,1,2,dp); } int dfs(vector<vector<int>>&graph,int t,int x,int y,vector<vector<vector<int>>>&dp){ if(t==graph.size()*2) return 0; if(x==y){ dp[t][x][y]=2;return 2; } if(x==0){ dp[t][x][y]=1;return 1; } if(dp[t][x][y]!=-1) return dp[t][x][y]; if(t%2==0){//老鼠走 bool flag=true; for(auto tt:graph[x]){ int now=dfs(graph,t+1,tt,y,dp); if(now==1){ dp[t][x][y]=1;return 1; } else if(now==0){ flag=false; } } if(flag){ dp[t][x][y]=2;return 2; } else{ dp[t][x][y]=0;return 0; } } else{//猫走 bool flag=true; for(auto tt:graph[y]){ if(tt==0) continue; int now=dfs(graph,t+1,x,tt,dp); if(now==2){ dp[t][x][y]=2;return 2; } else if(now==0){ flag=false; } } if(flag){ dp[t][x][y]=1;return 1; } else{ dp[t][x][y]=0;return 0; } } } };