200.Number of Islands(并查集C++)

里插入图片描述
题意大概就是有多少块1是连在一起的(上下左右),例1:所有1都是连在一起的所以是一块岛屿;
例2:四个1、一个1、两个1三块,所以答案是3;

class Solution {
public:
    vector<int> parent;
    vector<int> rank;   //记录树深度
    int count;    
    void init(vector<vector<char>>& grid){
            count=0;            
            int m=grid.size();            
            int n=grid[0].size();            
            for(int i=0;i<m;++i){                
            	for(int j=0;j<n;++j){                    
            	if(grid[i][j]=='1'){
            		parent.push_back(i*n+j);                        
            		count++;                    
          	 }                    
           	 else{                 
                	  parent.push_back(-1);                    
           	 }                    
                  rank.push_back(0);//每棵树的深度为0;            
          	}         
          }   
    }
    int find(int i){
          if (parent[i] != i) {
              parent[i] = find(parent[i]);    
          }
          return parent[i];
    }
    void unite(int x,int y){
    	int x_root=find(x);            
    	int y_root=find(y);            
    	if(x_root!=y_root){
    		if(rank[x_root]<rank[y_root]){//如果x的深度小就加在x结点下
    			swap(x_root,y_root);                
    		}            
    		parent[y_root]=x_root;
    		if(rank[x_root==y_root]) rank[x_root]+=1;
    		count--;
    	}
    }
    int numIslands(vector<vector<char>>& grid) {
    int nr=grid.size();        
    if(!nr) return 0;        
    int nc=grid[0].size();
    init(grid);//初始化parent数组
    	for(int i=0;i<nr;i++){
   		for(int j=0;j<nc;j++){
   			if(grid[i][j]=='1'){
   				grid[i][j]='0';                    
   				if(j+1<nc&&grid[i][j+1]=='1') unite(i*nc+j,i*nc+j+1);//向右找                    
   				if(i+1<nr&&grid[i+1][j]=='1') unite(i*nc+j,(i+1)*nc+j);//向下找
   			}
   		}
  	 }
   	return count;
   }
   };
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值