黑马程序员技术交流社区

标题: TreeSet [打印本页]
作者: 叶子和大人    时间: 2015-10-29 23:45
标题: TreeSet
TreeSet判断元素的惟一性的方法：compareTo方法的返回值，当返回值为0时，就是相同元素，不存。
//定义一个类，实现Comparable接口，重写接口的CompareTo方法。就可以让该类按照我们重写的方法进行判断元素相同与否。

  2. import java.util.*;
  3. class Person implements Comparable{
  4.         private String name;
  5.         private int age;
  6.         public Person(){}
  7.         public Person(String name,int age){
  8.                 this.name = name;
  9.                 this.age = age;
  10.         }
  11.         public void setName(String name){
  12.                 this.name = name;
  13.         }
  14.         public void setAge(int age){
  15.                 this.age = age;
  16.         }
  17.         public String getName(){
  18.                 return this.name;
  19.         }
  20.         public int getAge(){
  21.                 return this.age;
  22.         }
  23.         public int hashCode(){
  24.                 return name.hashCode()+age*23;
  25.         }
  26.         public boolean equals(Object obj)
  27.         {
  28.                 //传人两个相同的对象
  29.                 if(this == obj)
  30.                         return true;
  31.                 if(!(obj instanceof Person))
  32.                         throw new ClassCastException("类型错误");

  34.                 Person p = (Person)obj;
  35.                         return this.name.equals(p.name)&&this.age == p.age;
  36.        
  37.         }
  38.         public int compareTo(Object o){
  39.                 Person p = (Person)o;
  40.                 //先按照年龄排序，在按照姓名排序，以免年龄相同的人没有存进去
  41.                 int temp = this.age-p.age;
  42.                 return temp == 0?this.name.compareTo(p.name):temp;//这里的this.name是字符串，String类也实现了Comparable方法
  43.         }
  44. }

  46. public class TreeSetDemo{
  47.         public static void main(String[] args){
  48.                 TreeSet ts = new TreeSet();
  49.                 //Person对象年龄从小到大进行排序
  50.                 ts.add(new Person("zs",23));
  51.                 ts.add(new Person("ls",32));
  52.                 ts.add(new Person("ww",43));
  53.                 ts.add(new Person("zl",4));
  54.                 ts.add(new Person("tq",43));

  56.                 Iterator it = ts.iterator();
  57.                 while(it.hasNext()){
  58.                         Person p = (Person)it.next();
  59.                         System.out.println(p.getName()+"…"+p.getAge());
  60.                 }
  61.         }
  62. }
复制代码




欢迎光临 黑马程序员技术交流社区 (http://bbs.itheima.com/) 黑马程序员IT技术论坛 X3.2