题目:http://acm.hdu.edu.cn/showproblem.php?pid=3295
BFS+DFS,先任意选择一个颜色的点消去,然后查找这个点附近相邻的点一起消去。。这题查错。。被纠结死了。犯了很多错误。。DFS的时候把边界的判断放在了最后导致越界。。还有进入DFS时,没标记当前这个访问点。。导致内存爆掉。。= =!。。。代码打得太挫了。。
下面是AC代码:
#include<iostream>
#include<vector>
#include<queue>
#include<cstdlib>
#include<cstring>
#include<cstdio>
using namespace std;
int n,m;
int map[10][10];
bool vis[10][10];
int dir[4][2]= {{1,0},{0,1},{-1,0},{0,-1}};
struct node
{
vector<int > map[10];
int step;
} s_pos;
int ans,flag;
bool cheack(int x,int y)
{
return x>=0&&x<m&&y>=0&&y<n;
return false;
}
void dfs(int x,int y,int color,node &cur)
{
for(int i=0; i<4; i++)
{
int nx=x+dir[i][0],ny=y+dir[i][1];
if(cheack(nx,ny)&&!vis[nx][ny]&&cur.map[nx][ny]==color) //需要先检查边界!!!!
{
cur.map[nx][ny]=0;
vis[nx][ny]=true;
dfs(nx,ny,color,cur);
}
}
}
void move_zero(node &cur)
{
vector<int >temp[10];
int len=0,i,j;
for( i=0; i<m; i++)
{
for( j=0; j<n; j++)
if(cur.map[i][j]!=0) break;
if(j<n)
{
for( j=0; j<n; j++)
{
if(cur.map[i][j]!=0)
temp[len].push_back(cur.map[i][j]);
}
for(j=temp[len].size(); j<n; j++)
temp[len].push_back(0);
len++;
}
cur.map[i].clear();
}
for(int i=len; i<m; i++)
{
for(int j=0; j<n; j++)
temp[i].push_back(0);
}
for(int i=0; i<m; i++)
{
for(int j=0; j<n; j++)
{
cur.map[i].push_back(temp[i][j]);
}
}
}
bool find(node cur)
{
for(int i=0; i<m; i++)
for(int j=0; j<n; j++)
if(cur.map[i][j]!=0) return false;
return true;
}
void bfs()
{
queue<node >q;
q.push(s_pos);
while(!q.empty())
{
node now = q.front();
q.pop();
memset(vis,false,sizeof(vis));
if(find(now))
{
flag=1;
ans=now.step;
return ;
}
for(int i=0; i<m; i++)
{
for(int j=0; j<n; j++)
{
node next=now;
next.step+=1;
if(!vis[i][j]&&next.map[i][j]>0)
{
vis[i][j]=true;
dfs(i,j,next.map[i][j],next);
next.map[i][j]=0;
move_zero(next);
if(find(next))
{
flag=1;
ans=next.step;
return ;
}
q.push(next);
}
}
}
}
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=0; i<n; i++) for(int j=0; j<m; j++)
scanf("%d",&map[i][j]);
for(int i=0; i<m; i++) s_pos.map[i].clear();
s_pos.step=0;
for(int i=m-1; i>=0; i--) for(int j=n-1; j>=0; j--)
s_pos.map[i].push_back(map[j][i]);
flag=0;
if(find(s_pos)){
cout<<0<<endl;
continue ;
}
bfs();
if(flag) cout<<ans<<endl;
}
return 0;
}