这里dfs的话二维数组要剪枝(回溯)
class Solution {
public:
bool dfs(vector<vector<char>>& board, int i,int j,string word,int a){
if(a==word.size()||i<0||i>board.size()-1||j<0||j>board[0].size()-1||word[a]!=board[i][j])return false;
if(a==word.size()-1)return true;//而且判断相等不放在前面的话也会寄,可恶,这个破案例。
//flag[i][j]=false;
board[i][j]=' ';//剪支,用flag会超时不知道为啥。
bool res= dfs(board,i+1,j,word,a+1)||dfs(board,i,j+1,word,a+1)||dfs(board,i-1,j,word,a+1)||dfs(board,i,j-1,word,a+1);
board[i][j]=word[a];
return res;
}
bool exist(vector<vector<char>>& board, string word) {
if(board.size()==0)return false;
int k=word.size();
int n=board.size();
int m=board[0].size();
if(k>m*n)return false;
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if(board[i][j]==word[0]){//我的想法是先找第一个相等,结果这套代码好像不需要这样。
//vector<vector<bool>>flag(n,vector<bool>(m,true));
if(dfs(board,i,j,word,0))return true;
}
}
}
return false;
}
};
这道题剪枝是要影响大局的,所以用全局二维数组,并且二维数组不能出现在函数里(给萌新的,大佬当废话,最开始失误在这个地方)
class Solution {
public:
int count=0;
vector<vector<int>>res;
void dfs(int i,int j,int k){
if(i<0||i>=res.size()||j<0||j>=res[0].size()||(i/10+i%10+j/10+j%10)>k||res[i][j]==0)return;
count++;
res[i][j]=0;
dfs(i+1,j,k);
//dfs(i-1,j,k);
dfs(i,j+1,k);
//dfs(i,j-1,k);
}
int movingCount(int m, int n, int k) {
vector<vector<int>>resaa(m,vector<int>(n,1));
res=resaa;
dfs(0,0,k);
return count;
}
};