首页 > 代码库 > Set 的其他实现类：TreeSet

Set 的其他实现类：TreeSet

2024-09-08 03:35:26   215人阅读

  1. 向 TreeSet 中添加的元素必须是同一个类的
  2. 可以按照添加进集合中的元素的指定顺序遍历（仍然是“存储无序”的），如 String、包装类等默认按照从小到大的顺序遍历。
  3. 当向 TreeSet 中添加自定义类的对象时，有两种排序方法：自然排序、定制排序
  4. 自然排序：要求该类（要添加进 TreeSet 中的元素所对应的类）实现 java.lang.Comparable 接口（不实现的话不能添加）并实现其 compareTo(Object obj) 方法（非自定义类一般都已经实现了该方法），在该方法中指明按照自定义类的哪个属性进行排序的，该方法返回0时表示两元素用作排序的那个属性是相同的，返回负数代表当前元素的用作排序的那个属性是“字典式地”小于参数元素的用作排序的那个属性的，返回正数代表当前元素的用作排序的那个属性是“字典式地”大于参数元素的用作排序的那个属性的（在设计中首选那个不可能重复的属性来排序，以避免“A属性相同，再按B属性排序”的情况）
  5. 当向 TreeSet 中添加元素时，首先按照 compareTo( ) 进行比较，一旦返回0，虽然此时仅仅是指定的那个属性一样，但程序会认为两个对象是相同的，后面的对象不能被添加，为避免此情况发生，要求 compareTo( ) 与 equals( ) 和 hashCode( ) 保持一致
  6. 定制排序：调用TreeSet(Comparator comparator)  构造器构造 TreeSet 的对象，其中 comparator 是 java.util.Comparator 接口的实现类，并且该实现类的 compare(T o1, T o2)  方法指明了排序的根据，此时也要求 compare(T o1, T o2) 与 equals( ) 和 hashCode( ) 保持一致（理由同第5点）
一般不会让一个类即实现 Comparable 又定制 Comparator（这种情况以定制排序为准）

关于两种排序，参考一下代码：
  • 自然排序
  1. public class TestTreeSet {
  2. 	public static void main(String[] args) {
  3. 		TreeSet t = new TreeSet();
  4. 		t.add(new Person(12, "b"));
  5. 		t.add(new Person(12, "d"));
  6. 		t.add(new Person(11, "c"));
  7. 		t.add(new Person(10, "b"));
  8. 		System.out.println(t);// [Person [id=10, name=b],
  9. 								// Person [id=11, name=c],
  10. 								// Person [id=12, name=b],
  11. 								// Person [id=12, name=d]]
  13. 	}
  14. }
  16. class Person implements Comparable {
  17. 	private Integer id;
  18. 	private String name;
  20. 	public Person(int id, String name) {
  21. 		super();
  22. 		this.id = id;
  23. 		this.name = name;
  24. 	}
  26. 	public Person() {
  27. 		super();
  28. 	}
  30. 	public Integer getId() {
  31. 		return id;
  32. 	}
  34. 	public void setId(int id) {
  35. 		this.id = id;
  36. 	}
  38. 	public String getName() {
  39. 		return name;
  40. 	}
  42. 	public void setName(String name) {
  43. 		this.name = name;
  44. 	}
  46. 	@Override
  47. 	public int hashCode() {
  48. 		final int prime = 31;
  49. 		int result = 1;
  50. 		result = prime * result + id;
  51. 		result = prime * result + ((name == null) ? 0 : name.hashCode());
  52. 		return result;
  53. 	}
  55. 	@Override
  56. 	public boolean equals(Object obj) {
  57. 		if (this == obj)
  58. 			return true;
  59. 		if (obj == null)
  60. 			return false;
  61. 		if (getClass() != obj.getClass())
  62. 			return false;
  63. 		Person other = (Person) obj;
  64. 		if (id != other.id)
  65. 			return false;
  66. 		if (name == null) {
  67. 			if (other.name != null)
  68. 				return false;
  69. 		} else if (!name.equals(other.name))
  70. 			return false;
  71. 		return true;
  72. 	}
  74. 	@Override
  75. 	public String toString() {
  76. 		return "Person [id=" + id + ", name=" + name + "]";
  77. 	}
  79. 	@Override
  80. 	public int compareTo(Object o) {
  81. 		if (o instanceof Person) {
  82. 			Person p = (Person) o;
  83. 			int i = this.getId().compareTo(p.getId());
  84. 			if (i == 0) {
  85. 				return this.getName().compareTo(p.getName());
  86. 			}
  87. 			return i;
  88. 		}
  89. 		return 0;
  90. 	}
  91. }

  • 定制排序
  1. public class TestTreeSet {
  2. 	public static void main(String[] args) {
  3. 		TreeSet t = new TreeSet(new Comparator() {
  4. 			@Override
  5. 			public int compare(Object o1, Object o2) {
  6. 				if (o1 instanceof Person && o2 instanceof Person) {
  7. 					Person p1 = (Person) o1;
  8. 					Person p2 = (Person) o2;
  9. 					int i = p1.getId().compareTo(p2.getId());
  10. 					if (i == 0) {
  11. 						return p1.getName().compareTo(p2.getName());
  12. 					}
  13. 					return i;
  14. 				}
  15. 				return 0;
  16. 			}
  17. 		});
  18. 		t.add(new Person(12, "b"));
  19. 		t.add(new Person(12, "a"));
  20. 		t.add(new Person(11, "c"));
  21. 		t.add(new Person(10, "b"));
  22. 		System.out.println(t);// [Person [id=10, name=b],
  23. 								// Person [id=11, name=c],
  24. 								// Person [id=12, name=a],
  25. 								// Person [id=12, name=b]]
  27. 	}
  28. }
  30. class Person {
  31. 	private Integer id;
  32. 	private String name;
  34. 	public Person(int id, String name) {
  35. 		super();
  36. 		this.id = id;
  37. 		this.name = name;
  38. 	}
  40. 	public Person() {
  41. 		super();
  42. 	}
  44. 	public Integer getId() {
  45. 		return id;
  46. 	}
  48. 	public void setId(int id) {
  49. 		this.id = id;
  50. 	}
  52. 	public String getName() {
  53. 		return name;
  54. 	}
  56. 	public void setName(String name) {
  57. 		this.name = name;
  58. 	}
  60. 	@Override
  61. 	public int hashCode() {
  62. 		final int prime = 31;
  63. 		int result = 1;
  64. 		result = prime * result + id;
  65. 		result = prime * result + ((name == null) ? 0 : name.hashCode());
  66. 		return result;
  67. 	}
  69. 	@Override
  70. 	public boolean equals(Object obj) {
  71. 		if (this == obj)
  72. 			return true;
  73. 		if (obj == null)
  74. 			return false;
  75. 		if (getClass() != obj.getClass())
  76. 			return false;
  77. 		Person other = (Person) obj;
  78. 		if (id != other.id)
  79. 			return false;
  80. 		if (name == null) {
  81. 			if (other.name != null)
  82. 				return false;
  83. 		} else if (!name.equals(other.name))
  84. 			return false;
  85. 		return true;
  86. 	}
  88. 	@Override
  89. 	public String toString() {
  90. 		return "Person [id=" + id + ", name=" + name + "]";
  91. 	}
  92. }

Set 的其他实现类：TreeSet


联系
我们