首页 > 编程学习 > Java 8 HashMap(七)——TreeNode的getTreeNode()方法

说明看注释

        /**
         * 从根结点寻找h和k符合的结点
         */
        final TreeNode<K, V> getTreeNode(int h, Object k) {
            return ((parent != null) ? root() : this).find(h, k, null);
        }

        /**
         * 从根结点p开始根据hash和key值寻找指定的结点。kc是key的class
         *
         * @param h  hash值
         * @param k  key值
         * @param kc key的class
         */
        final TreeNode<K, V> find(int h, Object k, Class<?> kc) {
            //获得使用当前方法的节点p
            TreeNode<K, V> p = this;
            do {
                //获得当前节点的hash值 ph
                int ph = p.hash;
                //获得当前节点的Key pk
                K pk = p.key;
                //获得当前节点的左右子节点 pl pr
                TreeNode<K, V> pl = p.left, pr = p.right, q;
                int dir;
                if (ph > h) {
                    //如果p的hash值大于h,移动到p的左子节点
                    p = pl;
                } else if (ph < h) {
                    //如果p的hash值大于h,移动到p的右子节点
                    p = pr;
                } else if (pk == k || Objects.equals(k, pk)) {
                    //如果hash值相等,且k等于pk,说明找到p为要找的节点,返回p
                    return p;
                    //如果hash值相等但是key不相等,向非空的一边子节点移动
                } else if (pl == null) {
                    p = pr;
                } else if (pr == null) {
                    p = pl;
                } else if ((kc != null || (kc = comparableClassFor(k)) != null) &&
                        (dir = compareComparables(kc, k, pk)) != 0) {
                    //如果hash值相等但是key不相等,两边子节点都不为空
                    //当kc是可比较类比较k和pk,根据结果移动向左右子节点
                    p = (dir < 0) ? pl : pr;
                } else if ((q = pr.find(h, k, kc)) != null) {
                    //这里开始的条件仅当输入kc=null的时候才会进入,先检查右子树再检查左子树
                    return q;
                } else {
                    p = pl;
                }
            } while (p != null);
            return null;
        }

Copyright © 2010-2022 ngui.cc 版权所有 |关于我们| 联系方式| 豫B2-20100000