ReverseMap

用于Key和value都唯一的Map,方便根据value查找

public class ReverseMap<K, V> implements Map<K, V>{

    private Map<V, K> reverseMap;
    private Map<K, V> map;

    public ReverseMap(Class<? extends Map> clazz) throws Exception {
        map = (Map) clazz.newInstance();
        reverseMap = (Map)clazz.newInstance();
    }

    public ReverseMap(){
        map = new HashMap<>();
        reverseMap = new HashMap<>();
    }

    public Map<V, K> getReverseMap(){
        return reverseMap;
    }

    @Override
    public void clear() {
        map.clear();
        reverseMap.clear();
    }

    @Override
    public boolean containsKey(Object key) {
        return map.containsKey(key);
    }

    @Override
    public boolean containsValue(Object value) {
        return map.containsValue(value);
    }

    @Override
    public Set<Entry<K, V>> entrySet() {
        return map.entrySet();
    }

    @Override
    public V get(Object key) {
        return map.get(key);
    }

    public K getKey(Object value){
        return reverseMap.get(value);
    }

    @Override
    public boolean isEmpty() {
        return map.isEmpty();
    }

    @Override
    public Set<K> keySet() {
        return map.keySet();
    }

    public Set<V> valueSet(){
        return reverseMap.keySet();
    }

    @Override
    public V put(K key, V value) {
        if(reverseMap.containsKey(value)){
            throw new RuntimeException("have duplicates value " + value);
        }
        reverseMap.put(value, key);
        return map.put(key, value);
    }

    @Override
    public void putAll(Map<? extends K, ? extends V> map) {
        Map<? extends V, ? extends K> temp = CollectionUtil.reverseMap(map, true);
        this.map.putAll(map);
        this.reverseMap.putAll(temp);
    }

    @Override
    public V remove(Object key) {
        reverseMap.remove(map.get(key));
        return map.remove(key);
    }

    @Override
    public int size() {
        return map.size();
    }

    @Override
    public Collection<V> values() {
        return map.values();
    }
}
public class CollectionUtil {

    public static <K, V> Map<K, V> reverseMap(Map<V, K> map, boolean canReplace){

        Map<K, V> newMap = new HashMap<>();
        for(Map.Entry<V, K> entry : map.entrySet()){
            if(!canReplace && newMap.containsKey(entry.getValue())){
                throw new RuntimeException("have duplicates value " + entry.getValue());
            }
            newMap.put(entry.getValue(), entry.getKey());
        }
        return newMap;
    }

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值