码迷,mamicode.com
首页 > 其他好文 > 详细

UVA 572 搜索 初识BFS DFS

时间:2015-05-07 08:54:59      阅读:151      评论:0      收藏:0      [点我收藏+]

标签:搜索   dfs   bfs   stl   

题目类型: 搜索


样例输入:

1 1
*
3 5
*@*@*
**@**
*@*@*
1 8
@@****@*
5 5
****@
*@@*@
*@**@
@@@*@
@@**@
0 0


样例输出:

0
1
2
2

分析:

这一题可以说是搜索中最基础的一题之一。 要找出相连在一起的有多少块, 因此, 依次枚举,遇到@时就进行搜索,用深搜,广搜都行,目的是把相连的@都标记为已访问。

下面给出用DFS(深搜)和BFS(广搜)的代码

DFS  1 :递推

#include <iostream>
#include <cstring>
#include <cstdlib>
using namespace std;

int m,n,idx[105][105];
char pic[105][105];

void dfs(int r,int c,int id)
{
    if (r<0 || r>=m || c<0 || c>= n)   return ;
	if(idx[r][c] >0 || pic[r][c] != '@') return ;
	idx[r][c] = id;
	for (int dr = -1;dr <= 1;dr++)
	{
		for (int dc = -1;dc <= 1;dc++)
		{
			if(dr != 0 || dc != 0)
				dfs(r+dr,c+dc,id);
		}
	}
}

int main()
{
	int i,j,cnt;
	while (cin>>m>>n,m&&n)
	{
		for (i=0;i<m;i++)
		{
			cin>>pic[i];
		}
		cnt = 0;
		memset(idx,0,sizeof(idx));
        for (i=0;i<m;i++)
        {
			for (j=0;j<n;j++)
			{
               if(idx[i][j]==0 && pic[i][j] == '@')
				   dfs(i,j,++cnt);
			}
        }
		cout<<cnt<<endl;
	}
	return 0;
}
这种方法当数据量很大时,可能会有栈溢出
DFS  2:栈

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <stack>
using namespace std;

int m,n,idx[105][105];
char pic[105][105];
struct Node {int x;int y;};
stack<Node> st;

void dfs(int r,int c)
{
//     if (r<0 || r>=m || c<0 || c>= n)   return ;
// 	if(idx[r][c] >0 || pic[r][c] != '@') return ;
	while(!st.empty()) st.pop();
	Node temp;
	temp.x = r;
	temp.y = c;
	st.push(temp);
	idx[r][c] = 1;
	while (!st.empty())
	{
	
	temp = st.top();
    st.pop();
	for (int dr = -1;dr <= 1;dr++)
	{
		for (int dc = -1;dc <= 1;dc++)
		{
			int drr = dr+temp.x;
			int dcc = dc+temp.y;
			if((dr != 0 || dc != 0) && drr>=0 && drr<m && dcc>=0 && dcc<n  && pic[drr][dcc] == '@' && !idx[drr][dcc])
			//	dfs(r+dr,c+dc,id);
			{
			    Node t;
				idx[drr][dcc] =1;
				t.x = drr;
				t.y = dcc;
				st.push(t);
			}
		}
	}
	}
}

int main()
{
	int i,j,cnt;
	while (cin>>m>>n,m&&n)
	{
		for (i=0;i<m;i++)
		{
			cin>>pic[i];
		}
		cnt = 0;
		memset(idx,0,sizeof(idx));
        for (i=0;i<m;i++)
        {
			for (j=0;j<n;j++)
			{
               if(idx[i][j]==0 && pic[i][j] == '@')
			   {  
				   cnt++;
				   dfs(i,j);
			   }
			}
        }
		cout<<cnt<<endl;
	}
	return 0;
}

BFS:(队列实现)可用STL  也可以数组

#include <iostream>
#include <cstring>
#include <cstdlib>
#include <stack>
#include <queue>
using namespace std;

int m,n,idx[105][105];
char pic[105][105];
struct Node {int x;int y;};
//stack<Node> st;
//queue<Node> st;
Node que[105];

void dfs(int r,int c)
{
//     if (r<0 || r>=m || c<0 || c>= n)   return ;
// 	if(idx[r][c] >0 || pic[r][c] != '@') return ;
	//while(!st.empty()) st.pop();
	que[0].x = r;
	que[0].y = c;
	idx[r][c] = 1;
	int front=0;
	int rear=1;
	while (front < rear)
	{	
       Node temp=que[front++];
	for (int dr = -1;dr <= 1;dr++)
	{
		for (int dc = -1;dc <= 1;dc++)
		{
			int drr = dr+temp.x;
			int dcc = dc+temp.y;
			if((dr != 0 || dc != 0) && drr>=0 && drr<m && dcc>=0 && dcc<n  && pic[drr][dcc] == '@' && !idx[drr][dcc])
			{
				idx[drr][dcc] =1;
				que[rear].x = drr;
				que[rear].y = dcc;
				rear++;
			}
		}
	}
	}
}

int main()
{
	int i,j,cnt;
	while (cin>>m>>n,m&&n)
	{
		for (i=0;i<m;i++)
		{
			cin>>pic[i];
		}
		cnt = 0;
		memset(idx,0,sizeof(idx));
        for (i=0;i<m;i++)
        {
			for (j=0;j<n;j++)
			{
               if(idx[i][j]==0 && pic[i][j] == '@')
			   {  
				   cnt++;
				   dfs(i,j);
			   }
			}
        }
		cout<<cnt<<endl;
	}
	return 0;
}

UVA 572 搜索 初识BFS DFS

标签:搜索   dfs   bfs   stl   

原文地址:http://blog.csdn.net/xinwen1995/article/details/45542221

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!