Java用自定义的类型作为HashMap的key
Posted 杨铭宇
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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)));
}
}
运行结果为:
{[email protected]=zhangsan, [email protected]=wangwu, [email protected]=lisi, [email protected]=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;
}
}
重新执行上述函数,结果为:
{[email protected]=abc, [email protected]=zhangsan, [email protected]=lisi}
zhangsan
lisi
abc
可以看到,之前出的错误都被改正了
在HashMap中,查找key的比较顺序为:
- 计算对象的HashCode,看在表中是否存在
- 检查HashCode位置中的对象是否与当前对象相等
以上使用计算HashCode的方法在effective java第九点中提到:
- 对于对象中每个关键域f,为该域计算int类型的散列码c,result = 31 * result + c
以上是关于Java用自定义的类型作为HashMap的key的主要内容,如果未能解决你的问题,请参考以下文章