leetcode6-岛屿的个数--DFS或并查集

 思路：

采用DFS或者并查集方法

其中以下方法一和方法二转载自：https://blog.csdn.net/lv1224/article/details/82470100

方法一

采用DFS方式，对二维数据进行遍历，遇到1则找到一个岛屿，然后把与之关联的位置全部置为2，那么对二维数组完成一次遍历，即可得出结果。代码如下：

class Solution {
public:
    void setOne(vector<vector<char>> &grid,int row,int col){
        int m=grid.size();
        int n=grid[0].size();
        if((row<0)||(row>=m)||(col<0)||(col>=n)||(grid[row][col]!='1')){
            return;
        }
        grid[row][col]=2;
        setOne(grid,row+1,col);
        setOne(grid,row-1,col);
        setOne(grid,row,col+1);
        setOne(grid,row,col-1);
    }
    int numIslands(vector<vector<char>>& grid) {
        int m=grid.size();
        if(m==0) return 0;
        int n=grid[0].size();
        if(n==0) return 0;
        int sum=0;
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                if(grid[i][j]=='1'){
                    sum++;
                    setOne(grid,i,j);
                }
            }
        }
        return sum;
    }
};

方法二

也可以考虑用并查集来做这题，将二维数组重新编号，从0开始，从左到右，从上到下，直到n*m-1（其中n为行数，m为列数），对于位置(i,j)则编号为i*m+j，那么相邻（左右，上下）的为同一个值，则认为他们相通。那么最终只要统计一下father[i]==i且对应值为1的个数即可。代码如下：

class Solution {
public:
    int findFather(vector<int> &father,int a){
        int f=a;
        while(father[f]!=f){
            f=father[f];
        }
        while(father[a]!=a){
            int z=a;
            a=father[z];
            father[a]=f;

        }
        return f;
    }
    void unionFather(vector<int> &father, int a,int b){
        int fa=findFather(father,a);
        int fb=findFather(father,b);
        if(fa!=fb){
            father[fa]=fb;
        }
    }

    int numIslands(vector<vector<char>>& grid) {
        if(grid.size()==0||grid[0].size()==0) return 0;
        int n=grid.size(),m=grid[0].size(),k=n*m;
        vector<int> father(k,-1);
        for(int i=0;i<k;i++){
            father[i]=i;
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                int t1=i*m+j;
                int t2=t1+1,t3=t1+m;
                if(j+1<m&&grid[i][j]==grid[i][j+1]) unionFather(father,t1,t2);
                if(i+1<n&&grid[i][j]==grid[i+1][j]) unionFather(father,t1,t3);
            }
        }
        int res=0;
        for(int i=0;i<k;i++){
            if(father[i]==i&&grid[i/m][i%m]=='1'){
                res+=1;
            }
        }
        return res;
    }
};

 字节跳动第三次笔试又双叒出现这种题型了，第二题吧我记得是

好吧题目截的不全。。。

#include <iostream>
#include <string>
#include <map>
#include <vector>
using namespace std;

//找四周有没有1
void findOneNum(vector<vector<int>>&num,int M,int i,int j)
{
	if (i>=M||j>=M||i<0||j<0||num[i][j]!=1)
	{
		return;
	}
	
	num[i][j]=-1;
	
	findOneNum(num,M,i+1,j);
	findOneNum(num,M,i-1,j);
	findOneNum(num,M,i,j+1);
	findOneNum(num,M,i,j-1);
}

//DFS方式搜索获取个数
int GetNumPartner(vector<vector<int>>&num,int M)
{
	int outNum=0;
	for (int i=0;i<M;i++)
	{
		for (int j=0;j<M;j++)
		{
			if (num[i][j]==1)
			{
				outNum++;
				findOneNum(num,M,i,j);
			}
		}
	}
	return outNum;
}
int main()
{
	//第二题
	int M;cin>>M;
	vector<vector<int>>NumALL;
	for (int i=0;i<M;i++)
	{
		vector<int>num;
		for (int j=0;j<M;j++)
		{
			int temp;cin>>temp;
			num.push_back(temp);
		}
		NumALL.push_back(num);
	}
	int sum=GetNumPartner(NumALL,M);
	cout<<sum<<endl;
}

 完成80%系列。。要是有更好的欢迎讨论~~（不要问我为啥包含了这么多头文件，博主本地调试的时候是一个工程里写笔试几道题的西嘎嘎）