ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

算法第四版1.5union-find:习题1.5.17

2019-09-17 20:37:07  阅读:241  来源: 互联网

标签:count 1.5 17 int StdIn WeightedQuickUnionUF 习题 uf public


import edu.princeton.cs.algs4.StdIn;
import edu.princeton.cs.algs4.StdOut;

public class E1_5_17 {
    public static void main(String[]args){
        int N= StdIn.readInt();
        int cnt=RandomConnection.count(N);
        StdOut.printf("N=%8d  count=%15d\n",N,cnt);
    }
}

 

import edu.princeton.cs.algs4.StdRandom;

public class RandomConnection {
    public static int count(int N){
        //Union find.随机产生pair,直到所有点连在一起,返回产生的数的个数
        WeightedQuickUnionUF uf=new WeightedQuickUnionUF(N);
        int cnt=0;
        while (uf.count()!=1){
            int p= StdRandom.uniform(0,N);
            int q=StdRandom.uniform(0,N);
            uf.union(p,q);//函数中会判断是否已经连接
            cnt+=2;
        }
        return cnt;
    }
}
import edu.princeton.cs.algs4.StdIn;
import edu.princeton.cs.algs4.StdOut;

public class WeightedQuickUnionUF {
    public static void main(String[]args){
        //Solve dynamic connectivity problem on StdIn.
        int N= StdIn.readInt();
        WeightedQuickUnionUF uf=new WeightedQuickUnionUF(N);
        while (!StdIn.isEmpty()){
            int p=StdIn.readInt();
            int q=StdIn.readInt();
            if (uf.connected(p,q))continue;
            uf.union(p,q);
            StdOut.println(p+"  "+q);
        }
        StdOut.println(uf.count()+" components");
    }

    private int[]id; //parent link(site indexed)
    private int[]sz; //size of component for roots(site indexed) 只有根节点存储的是树的大小
    private int count;//number of component
    public WeightedQuickUnionUF(int N){
        count=N;
        id=new int[N];
        for (int i=0;i<N;i++)   id[i]=i;
        sz=new int[N];
        for (int i=0;i<N;i++)   sz[i]=1;
    }
    public int count(){
        return count;
    }
    public boolean connected(int p,int q){
        return find(p)==find(q);
    }
    public int find(int p){
        //Follow links to find a root.
        while (p!=id[p])p=id[p];
        return p;
    }
    public void union(int p,int q){
        int i=find(p);
        int j=find(q);
        if (i==q)return;
        //Make smaller root point to large one.
        if (sz[i]<sz[j]){id[i]=j;sz[j]+=sz[i];}
        else            {id[j]=i;sz[i]+=sz[j];}
        count--;
    }
}

 

标签:count,1.5,17,int,StdIn,WeightedQuickUnionUF,习题,uf,public
来源: https://blog.csdn.net/qq_40569107/article/details/100938678

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有