洛谷P1506(DFS染色问题)

P1506 拯救oibh总部

题目背景

oibh总部突然被水淹没了！现在需要你的救援……

题目描述

oibh被突来的洪水淹没了>.<还好oibh总部有在某些重要的地方起一些围墙，用号表示，而一个封闭的号区域洪水是进不去的……现在给出oibh的围墙建设图，问oibh总部没被淹到的重要区域（由"0"表示）有多少。

输入输出格式

输入格式：
第一行是两个数，x和y（x,y<=500）

第二行及以下是一个由和0组成的xy的图。

输出格式：
输出没被水淹没的oibh总部的“0”的数量。

这是一道染色问题,思路就是从边界(最外层那一圈)染色(染色为 * )进去，遇到*就回溯（return），最后找到有几个未染色的块。

#include <iostream>
#include<bits/stdc++.h>
using namespace std;
char a[505][505];
int X[4]={1,0,-1,0};
int Y[4]={0,1,0,-1};
int n,m;
bool check(int x,int y)
{
    if(x<0||y<0||x>=n||y>=m)///x->n,y->m
        return false;
    if(a[x][y]=='*') return false;
    return true;
}
void dfs(int x,int y)
{
    if(check(x,y)==false) return;///边界条件
    for(int i=0;i<4;i++)
    {
        a[x][y]='*';///染色
        int nx=x+X[i],ny=y+Y[i];
        dfs(nx,ny);
    }
}

int main()
{
    cin>>n>>m;getchar();
    int ans=0;
    for(int i=0;i<n;i++) for(int j=0;j<m;j++) cin>>a[i][j];
    for(int i=0;i<n;i++)///从边缘搜索进去
    {
        dfs(i,0);
        dfs(i,m-1);
    }
    for(int j=0;j<m;j++)///从边缘搜索进去
    {
        dfs(0,j);
        dfs(n-1,j);
    }
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(a[i][j]!='*')///未染色
                ans++;
        }
    }
    cout << ans << endl;
    return 0;
}