二维平面上的回溯算法(dfs)

回溯算法在二维平面上的搜索也能得到应用,也就是我们常说的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;
}

猜你喜欢

转载自blog.csdn.net/weixin_44316314/article/details/89526126