并查集的基础java实现

并查集的基本定义

import java.util.HashMap;
import java.util.List;
import java.util.Stack;

public class UnionFind {

    public static class Element<V> {
        public V value;

        public Element(V value) {
            this.value = value;
        }
    }

    public static class UnionFindSet<V> {
        //元素的map,获取节点内部包裹内容
        public HashMap<V, Element<V>> elementMap;
        // key 某个元素    value 元素的父
        public HashMap<Element<V>, Element<V>> fatherMap;
        // key 某个集合的代表元素    value 该集合的元素个数
        public HashMap<Element<V>, Integer> sizeMap;

        public UnionFindSet(List<V> list) {
            elementMap = new HashMap<>();
            fatherMap = new HashMap<>();
            sizeMap = new HashMap<>();

            for (V value :
                    list) {
                Element<V> element = new Element<>(value);
                elementMap.put(value, element);
                fatherMap.put(element, element); //刚开始自身成一个圈
                sizeMap.put(element, 1);
            }
        }

寻找代表元素方法


        //给定一个element 向上一直找 找到代表元素
        private Element<V> findHead(Element<V> element) {
            Stack<Element<V>> stack = new Stack<>();

            while (element != fatherMap.get(element)) {
                stack.push(element);
                element = fatherMap.get(element);
            }

            //将整个数据结构扁平化 变成一个高为2 的 多叉树
            while (!stack.isEmpty()) {
                fatherMap.put(stack.pop(), element);
            }

            return element;
        }

判断两个元素是不是一个集合

        //查询两个元素是不是一个集合
        public Boolean isSameSet(V a, V b) {
            if (elementMap.containsKey(a) && elementMap.containsValue(b)) {
                return findHead(elementMap.get(a)) == findHead(elementMap.get(b));
            }
            return false;
        }

并查集合并方法


        public void union(V a, V b) {
            if (elementMap.containsKey(a) && elementMap.containsValue(b)) {
                Element<V> afather = findHead(elementMap.get(a));
                Element<V> bfather = findHead(elementMap.get(b));

                if (afather != bfather) {
                    Element<V> big = sizeMap.get(afather) >= sizeMap.get(bfather) ? afather : bfather;
                    Element<V> small = big == afather ? bfather : afather;
                    fatherMap.put(small,big);
                    sizeMap.put(big,sizeMap.get(afather)+sizeMap.get(bfather));
                }
            }
        }
    }


}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
您可以使用以下步骤在Java中查询PostgreSQL数据库中的树: 1. 首先,您需要连接到PostgreSQL数据库。您可以使用JDBC驱动程序来实现这一点。以下是一个示例代码片段: ```java String url = "jdbc:postgresql://localhost:5432/mydb"; String user = "myuser"; String password = "mypassword"; Connection conn = DriverManager.getConnection(url, user, password); ``` 2. 接下来,您需要编写SQL查询来获取树的数据。以下SQL查询是一个示例: ```sql WITH RECURSIVE tree(id, name, parent_id, level, path) AS ( SELECT id, name, parent_id, 0, ARRAY[id] FROM mytable WHERE parent_id IS NULL UNION ALL SELECT c.id, c.name, c.parent_id, p.level + 1, p.path || c.id FROM mytable c JOIN tree p ON c.parent_id = p.id ) SELECT id, name, parent_id, level, path FROM tree ORDER BY path; ``` 此查询使用了递归公共表达式(WITH RECURSIVE),它将每个节点的ID、名称、父节点ID、级别和路径存储在结果集中。路径列是一个整数数组,其中包含每个节点的ID。您可以根据需要修改此查询以适应您的数据模型。 3. 在Java中执行查询并处理结果集。以下是一个示例: ```java PreparedStatement stmt = conn.prepareStatement(query); ResultSet rs = stmt.executeQuery(); while (rs.next()) { int id = rs.getInt("id"); String name = rs.getString("name"); Integer parentId = rs.getInt("parent_id"); int level = rs.getInt("level"); Array path = rs.getArray("path"); // process node data here } rs.close(); stmt.close(); conn.close(); ``` 在此示例中,我们使用PreparedStatement执行SQL查询,并使用ResultSet迭代结果集中的每个节点。使用ResultSet的方法获取每个节点的ID、名称、父节点ID、级别和路径。您可以根据需要修改此代码以适应您的数据模型。 以上是在Java中查询PostgreSQL数据库中的树的基本步骤。根据您的具体需求,您可能需要在此基础上进行更多的调整和优化。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值