Java用自定义的类型作为HashMap的key

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

需要重写hashCode()和equals()方法才可以实现自定义键在HashMap中的查找。

public class PhoneNumber

{

    private int prefix; //区号

    private int phoneNumber; //电话号

    public PhoneNumber(int prefix, int phoneNumber)

    {

        this.prefix = prefix;

        this.phoneNumber = phoneNumber;

    }

}

import java.util.HashMap;

public class Test1

{

    public static void main(String[] args)

    {

        HashMap<PhoneNumber, String> map = new HashMap<>();

        map.put(new PhoneNumber(027, 12345678), "zhangsan");

        map.put(new PhoneNumber(027, 22222222), "lisi");

        map.put(new PhoneNumber(027, 33333333), "wangwu");

        map.put(new PhoneNumber(027, 33333333), "abc");

        System.out.println(map.toString());

        System.out.println(map.get(new PhoneNumber(027, 12345678)));

        System.out.println(map.get(new PhoneNumber(027, 22222222)));

        System.out.println(map.get(new PhoneNumber(027, 33333333)));

    }

}

运行结果为：

{package1.PhoneNumber@74a14482=zhangsan, package1.PhoneNumber@677327b6=wangwu, package1.PhoneNumber@1540e19d=lisi, package1.PhoneNumber@14ae5a5=abc}

null

null

null

从中我们可以看到出现了两个问题：

new PhoneNumber(027, 33333333)这个键被添加了两次，但是在HashMap中wangwu和abc同时存在了
使用get方法取得的值均为null

正确的方法就是直接对PhoneNumber类进行修改，覆盖equals和hashCode方法，修改后的PhoneNumber类如下：

public class PhoneNumber

{

    private int prefix; //区号

    private int phoneNumber; //电话号

    public PhoneNumber(int prefix, int phoneNumber)

    {

        this.prefix = prefix;

        this.phoneNumber = phoneNumber;

    }

    @Override

    public boolean equals(Object o)

    {

        if(this == o)

        {

            return true;

        }

        if(!(o instanceof PhoneNumber))

        {

            return false;

        }

        PhoneNumber pn = (PhoneNumber)o;

        return pn.prefix == prefix && pn.phoneNumber == phoneNumber;

    }

    @Override

    public int hashCode()

    {

        int result = 17;

        result = 31 * result + prefix;

        result = 31 * result + phoneNumber;

        return result;

    }

}