力扣 959. 由斜杠划分区域 遍历/并查集

https://leetcode-cn.com/problems/regions-cut-by-slashes/
力扣 959. 由斜杠划分区域 遍历/并查集_第1张图片
思路:重点在于怎么把输入转换成图,我们可以把每个字符抽象为3*3的方格,认为斜杠所处的方格不能走,那么这个图的连通分量个数就是答案。并查集的思路也和这个类似,把每个字符抽象为4个拼在一起的三角形(实际上还是一个方格),然后对斜杠分类讨论即可,感兴趣的话可以看一下官方题解。

class Solution {
     
public:

    int d[4][2]={
     {
     0,1},{
     0,-1},{
     1,0},{
     -1,0}};

    void dfs(vector<vector<bool>> &grid,int x,int y)
    {
     
        grid[x][y]=1;
        int n=grid.size();
        for(int i=0;i<4;i++)
        {
     
            int dx=x+d[i][0];
            int dy=y+d[i][1];
            if(dx>=0&&dy>=0&&dx<n&&dy<n&&!grid[dx][dy])
                dfs(grid,dx,dy);
        }
    }

    int regionsBySlashes(vector<string>& grid) {
     
        int n=grid.size();
        if(!n)
            return n;
        vector<vector<bool>> vec(3*n,vector<bool>(3*n,0));
        for(int i=0;i<n;i++)
        {
     
            for(int j=0;j<n;j++)
            {
     
                int i3=i*3,j3=j*3;
                if(grid[i][j]=='/')
                    vec[i3][j3+2]=vec[i3+1][j3+1]=vec[i3+2][j3]=1;
                else if(grid[i][j]=='\\')
                    vec[i3][j3]=vec[i3+1][j3+1]=vec[i3+2][j3+2]=1;
            }
        }
        int ans=0;
        n*=3;
        for(int i=0;i<n;i++)
        {
     
            for(int j=0;j<n;j++)
            {
     
                if(!vec[i][j])
                {
     
                    ++ans;
                    dfs(vec,i,j);
                }
            }
        }
        return ans;
    }
};

你可能感兴趣的:(力扣,DFS,并查集)