回溯算法在二维平面上的搜索也能得到应用,也就是我们常说的dfs;
题目
给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
输入:
3 4
ABCE
SFCS
ADEE
给定 word = “ABCCED”, 返回 true.
给定 word = “SEE”, 返回 true.
给定 word = “ABCB”, 返回 false.
#include <iostream>
#include <string>
using namespace std;
string a[100],word;
int n,m;
int d[4][2]={{-1,0},{1,0},{0,1},{0,-1}},visited[100][100]={0};
//从a[startx][starty]开始搜索word[index...word.size()-1];
bool SearchWord( int index,int startx,int starty )
{
if( startx < 0 || startx >= n || starty <0 || starty >=m )
return false;
if( index == word.size()-1 )
{
return a[startx][starty] == word[index];
}
if( a[startx][starty] == word[index] )
{
//从startx,starty出发,向四个方面搜索;
for( int i = 0 ; i < 4 ; i++ )
{
int newx = startx + d[i][0];
int newy = starty + d[i][1];
if( !visited[newx][newy] )
{
visited[newx][newy] = 1;
if( SearchWord( index+1,newx,newy ) )
return true;
visited[newx][newy] = 0;
}
}
}
return false;
}
int main()
{
cin >> n >> m;
for( int i = 0 ; i < n ; i++ )
{
cin >> a[i];
}
cin >> word;
for( int i = 0 ; i< n ; i++ )
{
for( int j = 0 ; j < m ; j++ )
{
if( SearchWord( 0,i,j ) )
{
cout << "true" << endl;
return 0;
}
}
}
cout << "false" << endl;
return 0;
}