POJ 1979:Red and Black

新博客第一篇文章，题目按照《挑战程序设计竞赛》一书来练习，希望可以有提高。

问题描述

给出有红砖与黑砖组成的图以及起点，黑砖可以走，红砖不可走，问最多可以遍历多少黑砖。

解题思路

很简单的DFS。

代码

#include <iostream>#include <stdio.h>using namespace std;char room[20][20];int w,h;int sum;void dfs(int i,int j){    sum++;    room[i][j]='#';    if((i!=0)&&(room[i-1][j]=='.'))        dfs(i-1,j);    if((i!=(h-1))&&(room[i+1][j]=='.'))        dfs(i+1,j);    if((j!=0)&&(room[i][j-1]=='.'))        dfs(i,j-1);    if((j!=(w-1))&&(room[i][j+1]=='.'))        dfs(i,j+1);}int main(){    freopen("in.txt","r",stdin);    while(cin>>w>>h)    {        if((w==0)&&(h==0))            break;        int begini,beginj;        sum=0;        for(int i=0;i<h;i++)            for(int j=0;j<w;j++)            {                 cin>>room[i][j];                 if(room[i][j]=='@')                 {                     begini=i;                     beginj=j;                 }            }        dfs(begini,beginj);        cout<<sum<<endl;    }    return 0;}

问题与反思

输入的时候w与h写反了，出了一点小bug，改完之后一次AC了。