话题:重写HashSet中的equals()方法和hashCode()方法
equals方法常用在String类中比较两个字符串是否同,但是在其他的类中也有着equals方法。比如HashSet,我们知道HashSet中是没有重复对象的,这是因为它的add方法在添加集合对象之前会自动调用hashCode方法判断即将添加的对象是否已经存在于集合中,也就是判断集合中是否存在与即将添加对象的hash值相等的对象。如果存在则不添加,不存在则调用equals方法继续判断当前集合中是否存在和即将添加对象相同的对象,存在则不添加,不存在则添加。
这看起来很美好,但是在现实编码的时候我们往HashSet集合中添加的对象是不确定的,如果添加一个Student类对象,它有name,age这两个属性,equals方法要判断添加进集合的每一个Student对象的name,age两个属性是否已经存在于集合。添加一个Person类对象,它有name,age,gender这三个属性,add方法要判断添加进集合的每一个Person对象的name,age,gender这三个属性是否已经存在于集合。这不可能实现,因为添加对象的类型,属性不确定,并不能做到具体问题具体分析。
如果不重写equals方法,仅仅依靠hashCode方法去判断添加对象是否已存在集合:
public class HashSetDemo {
public static void main(String[] args) {
HashSet<Student> hs = new HashSet<Student>();
Student s1 = new Student("zhangsan",19);
hs.add(s1);
hs.add(s1);
hs.add(s1);
for (Student student : hs) {
System.out.println(student.toString());
}
}
}
class Student {
String name;
int age;
public Student(String name,int age) {
this.name = name;
this.age = age;
}
@Override
public String toString() {
return "Student [name=" + name + ", age=" + age + "]";
}
}
上述代码执行结果:
看着好像也能实现简单的判断去重(我们往hs里面添加了三次s1,但是hs中只有一个s1)。那是因为添加的对象hash值相同,都是s1的hash值。
但是如果我们在上述代码的基础上加以改动,如下:
public class HashSetDemo {
public static void main(String[] args) {
HashSet<Student> hs = new HashSet<Student>();
Student s1 = new Student("zhangsan",19);
Student s2 = new Student("zhangsan",19);//在new一个zhangsan,19
hs.add(s1);
hs.add(s1);
hs.add(s2);//将zhangsan,19添加进hs
for (Student student : hs) {
System.out.println(student.toString());
}
}
}
class Student {
String name;
int age;
public Student(String name,int age) {
this.name = name;
this.age = age;
}
@Override
public String toString() {
return "Student [name=" + name + ", age=" + age + "]";
}
}
改动后的代码执行结果如下:
很明显,hs中有两个对象,s1和s2,因为s2和s1的hash值不同,所以add方法调用的hashCode方法判断hs中不存在s2,于是进行了添加。但这不是我们想要的,因为你添加的两个学生都叫zhangsan,年纪也都是19,这样一来hs中还是有重复的元素存在,这样说明add方法调用的hashCode方法根据判断hash值的不同来确定是否要添加该对象时是很容易出问题的,是不准确的。还好在判断hash值之后,add方法还会调用equals方法判断对象的属性。因此我们重写equals方法。
重写equals方法得具体问题具体分析,针对添加的对象不同,重写不同的equals方法,添加student类对象,equals判断姓名,年龄。添加Person对类对象,equals判断hash值,姓名,年龄,性别。。。。这样才能精准的实现HashSet去重。
重写equals方法实现去重,拿添加Person类对象来分析:
(1)首先我们在Person中重写toString方法,让它按照我们想让它输出的格式输出。
@Override
public String toString() {
return "PersonTest [name=" + name + ", age=" + age + "]";
}
(2)因为都是同一个Person类对象,我们就连hashCode方法也一块重写了,让它返回同一个hash值就行,我们只比较姓名和年龄是否相同,不同才看做为不同的对象。我们就随便返回一个int值即可。
@Override
public int hashCode() {
return 1;
}
(3)接下来才是是equals方法,我们做5种情况的判断
①因为我们添加的是一个对象,所以参数是一个对象,如果这个对象和当前作为参照的对象是同一个对象,则相同,看作存在,返回true,不添加。
if (this == obj) {
return true;
}
②如果添加的对象和当前类的类名不是同一个类名,添加,返回false
if (this.getClass() != obj.getClass()) {
return false;
}
③因为传进来的是不确定类的对象,用泛型参数接收。所以对比元素的时候,先向下转型为Person类。如果当前对象的name为null,添加对象的名字不为null,返回false,添加。或者当前对象name不等于添加对象name,返回false,添加。
PersonTest p = (PersonTest) obj;
if (name == null) {
if (p.name != null) {
return false;
} else if (this.name != p.name) {
return false;
}
}
④如果当前对象的名字和添加对象的名字不相等,返回false,添加。
if (!this.name.equals(p.name)) {
return false;
}
⑤判断当前对象的age是否和添加对象的age相等。
if (this.age != p.age) {
return false;
}
⑥如果上面都不满足,返回true,不添加。return true;
因此完整的针对举例的Person类对象添加进HashSet集合重写equals方法的代码如下:(测试类代码可自己写)
public class Person {
String name;
int age;
public Person(String name, int age) {
this.name = name;
this.age = age;
}
@Override
public String toString() {
return "Person [name=" + name + ", age=" + age + "]";
}
@Override
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (this.getClass() != obj.getClass()) {
return false;
}
if (obj == null) {
return false;
}
PersonTest p = (Person) obj;
if (name == null) {
if (p.name != null) {
return false;
} else if (this.name != p.name) {
return false;
}
}
if (!this.name.equals(p.name)) {
return false;
}
if (this.age != p.age) {
return false;
}
return true;
}
@Override
public int hashCode() {
return 1;
}
}
结语:后面分析代码那一段写的有点乱。
其实拿eclipse来说,有自带的重写equals方法,能自动生成。
右击 -->Source -->Generate hashCode() and equals() 能自动生成方法。哈哈。
如果您觉得本文有什么错误的话欢迎您评论指点?,如果本文对您有所帮助的话小弟也非常荣幸!如果您愿意和小弟交流学习的话可以点击关注,我们可以一起学习一起进步?。