TreeSet判断重复元素
2017-11-03 10:25
323 查看
TreeSet的底层是TreeMap的keySet(),而TreeMap是基于红黑树实现的,红黑树是一种平衡二叉查找树,它能保证任何一个节点的左右子树的高度差不会超过较矮的那棵的一倍。
TreeMap是按key排序的,所以TreeSet中的元素也是排好序的。显然元素在插入TreeSet时compareTo()方法要被调用,所以TreeSet中的元素要实现Comparable接口。TreeSet作为一种Set,它不允许出现重复元素。TreeSet是用compareTo()来判断重复元素的,而非equals(),看下面代码。
import java.util.TreeSet;
import org.junit.Test;
public class TestTreeSet {
class Combine implements Comparable<Combine> {
private int p1;
private int p2;
public Combine(int p1, int p2) {
this.p1 = p1;
this.p2 = p2;
}
@Override
public int hashCode() {
return p1 * 31 + p2;
}
@Override
public boolean equals(Object obj) {
System.out.print("whether equal " + this + " and " + obj);
boolean rect = false;
if (obj instanceof Combine) {
System.out.println("whether equal " + this + " and " + obj);
Combine other = (Combine) obj;
rect = (this.p1 == other.getP1() && this.p2 == other.getP2());
}
System.out.println(": " + rect);
return rect;
}
@Override
public int compareTo(Combine o) {
System.out.print("compare " + this + " and " + o);
// 排序时只考虑p1
if (this.p1 < o.p1) {
System.out.println(", return -1");
return -1;
} else if (this.p1 > o.p1) {
System.out.println(", return 1");
return 1;
} else {
System.out.println(", return 0");
return 0;
}
}
@Override
public String toString() {
return "(" + p1 + "," + p2 + ")";
}
public int getP1() {
return p1;
}
public void setP1(int p1) {
this.p1 = p1;
}
public int getP2() {
return p2;
}
public void setP2(int p2) {
this.p2 = p2;
}
}
@Test
public void test() {
Combine c1 = new Combine(1, 2);
Combine c2 = new Combine(1, 2);
Combine c3 = new Combine(1, 3);
Combine c4 = new Combine(5, 2);
TreeSet<Combine> set = new TreeSet<Combine>();
set.add(c1);
set.add(c2);
set.add(c3);
set.add(c4);
while (!set.isEmpty()) { //按顺序输出TreeSet中的元素
Combine combine = set.pollFirst();
System.out.println(combine.getP1() + "\t" + combine.getP2());
}
}
}
输出:
compare (1,2) and (1,2), return 0
compare (1,2) and (1,2), return 0
compare (1,3) and (1,2), return 0
compare (5,2) and (1,2), return 1
1 2
5 2
我们看到不论compareTo()返回的是不是相等,equals()方法都没有被调用。
转载地址
TreeMap是按key排序的,所以TreeSet中的元素也是排好序的。显然元素在插入TreeSet时compareTo()方法要被调用,所以TreeSet中的元素要实现Comparable接口。TreeSet作为一种Set,它不允许出现重复元素。TreeSet是用compareTo()来判断重复元素的,而非equals(),看下面代码。
import java.util.TreeSet;
import org.junit.Test;
public class TestTreeSet {
class Combine implements Comparable<Combine> {
private int p1;
private int p2;
public Combine(int p1, int p2) {
this.p1 = p1;
this.p2 = p2;
}
@Override
public int hashCode() {
return p1 * 31 + p2;
}
@Override
public boolean equals(Object obj) {
System.out.print("whether equal " + this + " and " + obj);
boolean rect = false;
if (obj instanceof Combine) {
System.out.println("whether equal " + this + " and " + obj);
Combine other = (Combine) obj;
rect = (this.p1 == other.getP1() && this.p2 == other.getP2());
}
System.out.println(": " + rect);
return rect;
}
@Override
public int compareTo(Combine o) {
System.out.print("compare " + this + " and " + o);
// 排序时只考虑p1
if (this.p1 < o.p1) {
System.out.println(", return -1");
return -1;
} else if (this.p1 > o.p1) {
System.out.println(", return 1");
return 1;
} else {
System.out.println(", return 0");
return 0;
}
}
@Override
public String toString() {
return "(" + p1 + "," + p2 + ")";
}
public int getP1() {
return p1;
}
public void setP1(int p1) {
this.p1 = p1;
}
public int getP2() {
return p2;
}
public void setP2(int p2) {
this.p2 = p2;
}
}
@Test
public void test() {
Combine c1 = new Combine(1, 2);
Combine c2 = new Combine(1, 2);
Combine c3 = new Combine(1, 3);
Combine c4 = new Combine(5, 2);
TreeSet<Combine> set = new TreeSet<Combine>();
set.add(c1);
set.add(c2);
set.add(c3);
set.add(c4);
while (!set.isEmpty()) { //按顺序输出TreeSet中的元素
Combine combine = set.pollFirst();
System.out.println(combine.getP1() + "\t" + combine.getP2());
}
}
}
输出:
compare (1,2) and (1,2), return 0
compare (1,2) and (1,2), return 0
compare (1,3) and (1,2), return 0
compare (5,2) and (1,2), return 1
1 2
5 2
我们看到不论compareTo()返回的是不是相等,equals()方法都没有被调用。
转载地址
相关文章推荐
- HashSet与TreeSet对重复元素的判断不同之处
- HashSet与TreeSet对重复元素的判断不同之处
- TreeSet判断重复元素解析及代码示例
- 链表有环判断,快慢指针两种方法/合并链表/删除重复元素/二分递归和while
- 判断Set里的元素是否重复、==、equals、hashCode方法研究-代码演示
- hashCode方法的作用----java判断来年各个元素是否重复的机制
- 判断一个数组中的元素是否重复?
- 217. Contains Duplicate (判断数组中是否有重复元素)
- Set判断重复,TreeSet排序
- HashSet中是如何判断元素是否重复的
- C#实现如何判断一个数组中是否有重复的元素
- 一个长度为N的整形数组,数组中每个元素的取值范围是[0,N-1],写一个算法判断数组中是否存在重复的数字
- js 判断数组是否有重复元素的几种方式
- HashSet中是如何判断元素是否重复的
- 219. Contains Duplicate II (判断数组中是否有重复元素且其下标之差不超过k)
- 如何判断一个整数数组中是否有重复元素?要求时间复杂度O(n),空间复杂度O(1)
- 12、一个整数数列,元素取值可能是0~65535中的任意一个数,相同数值不会重复出现。0是例外,可以反复出现。 请设计一个算法,当你从该数列中随意选取5个数值,判断这5个数值是否连续相邻。
- js中判断数组中的元素是否重复
- 如何判断一个整数数组中是否有重复元素?要求时间复杂度O(n),空间复杂度O(1)
- JavaScript 判断两个数组是否相等 或者 判断一个数组中是否有重复的元素