Floodfill求连通分量

floodfill的基本思路就是找到所有连在一起的结点并染成相同的颜色，这样就可以通过颜色区分各个连通分量。具体来说，一开始把所有结点都设成相同的颜色，然后从第一个结点往后遍历，遇到还未被染成其他色的结点，表明这个结点不属于任何已知的连通分量，所以给它一个新颜色，并用bfs/dfs给与其相连的所有结点都染成这个颜色。

代码如下：

//邻接链表实现floodfill
class Graph {
private:
    int n;
    int *color;
    vector<int> *edges;
    
public:
    Graph(int input_n) {
        n = input_n;
        edges = new vector<int>[n];
        color = new int[n];
        //set all color to be the 0 at first
        memset(color, 0, n * sizeof(int));
    }

    ~Graph() {
        delete[] edges;
        delete[] color;
    }

    void insert(int x, int y) {
        edges[x].push_back(y);
        edges[y].push_back(x);
    }

    void floodfill() {
        int color_cnt=0;
        //bfs for every node as starting node
        for(int i=0;i<n;i++){
            //if color==0 means it is seperate from previous ones
            //so should start a new connected component
            if(color[i]==0){
                //update color for this component
                color_cnt++;
                color[i]=color_cnt;
                queue<int> q;
                //this node will be the "root node" for this component
                q.push(i);
                while(!q.empty()){
                    int vertex=q.front();
                    for(int j: edges[vertex]){
                        if(color[j]==0){
                            color[j]=color_cnt;
                            q.push(j);
                        }
                    }
                    q.pop();
                }
            }
        }
        for(int i=0;i<n;i++){
            cout<<i<<" "<<color[i]<<endl;
        }
    }
};

 

这里再给一个例子，大意就是找出上下左右相连通的连通块(‘*')的个数：

//floodfill求连通块个数
#include <iostream>
#include <cstring>
using namespace std;

int dir[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
char map[1001][1001];
int color[1001][1001];
int height,width;

void dfs(int x, int y, int cur_color){
    int curx, cury;
    for(int i=0;i<4;i++){
        curx=x+dir[i][0];
        cury=y+dir[i][1];
        if(map[curx][cury]=='*'&&color[curx][cury]==0){
            color[curx][cury]=cur_color;
            dfs(curx,cury,cur_color);
        }
    }
}

int floodfill(){
    int color_cnt=0;
    for(int i=1;i<=height;i++){
        for(int j=1;j<=width;j++){
            if(map[i][j]=='*'&&color[i][j]==0){
                color_cnt++;
                dfs(i,j,color_cnt);
            }
        }
    }
    return color_cnt;
}


int main(){
    cin>>height>>width;
    for(int i=0;i<=height+1;i++){
        for(int j=0;j<=width+1;j++){
            map[i][j]='x';
        }
    }
    for(int i=1;i<=height;i++){
        for(int j=1;j<=width;j++){
            cin>>map[i][j];
        }
    }
    memset(color,0,sizeof(color));
    cout<<floodfill()<<endl;
}