如何获取LinkedHashMap中指定键对应元素的下一个元素？

LinkedHashMap的内部Entry节点维护了before和after指针来保证插入顺序，因此可以直接利用这一特性跳过遍历，直接定位目标键的下一个元素，以下是两种可行方案，优先推荐第一种：

方案一：自定义LinkedHashMap子类（推荐） #

通过继承LinkedHashMap，暴露获取指定Entry和其下一个Entry的方法，完全合法且无封装破坏风险，性能最优。

代码实现 #

// 自定义子类，暴露获取Entry及下一个Entry的方法
class AccessibleLinkedHashMap<K, V> extends LinkedHashMap<K, V> {
    // 获取指定键对应的Entry
    public Entry<K, V> getTargetEntry(K key) {
        return super.getEntry(key);
    }

    // 获取指定Entry的下一个Entry
    public Entry<K, V> getNextEntry(Entry<K, V> entry) {
        return entry.after;
    }
}

使用示例 #

public static void main(String[] args) {
    AccessibleLinkedHashMap<Integer, String> map = new AccessibleLinkedHashMap<>();
    map.put(10, "C");
    map.put(20, "C++");
    map.put(50, "JAVA");
    map.put(40, "PHP");
    map.put(30, "Kotlin");

    // 获取键50对应的Entry
    LinkedHashMap.Entry<Integer, String> targetEntry = map.getTargetEntry(50);
    if (targetEntry != null) {
        LinkedHashMap.Entry<Integer, String> nextEntry = map.getNextEntry(targetEntry);
        if (nextEntry != null) {
            System.out.println("下一个元素：键=" + nextEntry.getKey() + ", 值=" + nextEntry.getValue());
            // 输出：下一个元素：键=40, 值=PHP
        } else {
            System.out.println("该元素已是最后一个，无下一个元素");
        }
    } else {
        System.out.println("指定键不存在");
    }
}

方案二：反射方式（不推荐） #

若无法自定义子类，可通过反射获取LinkedHashMap内部的Entry及after字段，但该方式破坏封装，依赖JDK内部实现，版本兼容性差，仅作为应急方案。

代码实现 #

import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.LinkedHashMap;
import java.util.Map;

public class Test {
    public static void main(String[] args) throws Exception {
        Map<Integer, String> map = new LinkedHashMap<>();
        map.put(10, "C");
        map.put(20, "C++");
        map.put(50, "JAVA");
        map.put(40, "PHP");
        map.put(30, "Kotlin");

        // 反射调用getEntry方法获取目标键的Entry
        Method getEntryMethod = LinkedHashMap.class.getDeclaredMethod("getEntry", Object.class);
        getEntryMethod.setAccessible(true);
        LinkedHashMap.Entry<Integer, String> targetEntry = 
            (LinkedHashMap.Entry<Integer, String>) getEntryMethod.invoke(map, 50);

        if (targetEntry != null) {
            // 反射获取Entry的after字段
            Field afterField = LinkedHashMap.Entry.class.getDeclaredField("after");
            afterField.setAccessible(true);
            LinkedHashMap.Entry<Integer, String> nextEntry = 
                (LinkedHashMap.Entry<Integer, String>) afterField.get(targetEntry);

            if (nextEntry != null) {
                System.out.println("下一个元素：键=" + nextEntry.getKey() + ", 值=" + nextEntry.getValue());
            } else {
                System.out.println("该元素已是最后一个");
            }
        }
    }
}

总结 #

单线程场景下，自定义LinkedHashMap子类是最优方案，无需遍历集合，直接通过内部指针定位下一个元素，性能最高且安全可靠。

内容的提问来源于stack exchange，提问作者user20301714