黑马程序员_<<Set,TreeSet>>

--------------------ASP.Net+Android+IOS开发、.Net培训、期待与您交流！ --------------------

1.TreeSet

1. 概述

TreeSet本身对元素记性排序，要是自定的类，那么要是此类对象存数TreeSet中，那么就必须是S自定义的类本身具备比较性，那么据必须实现接口Comparable，并重写方法compareTo()方法,底层数据结构是二叉树

compareTo()返回值是整数，如果小于0，那么此对象就小于比较的对象，等于0，那么此对象就等于比较的对象，如果大于0，那么此对象就大于比较的对象。

2. 自然排序规则(Comparable)

有的对象在存储到TreeSet本身就具备可比性，例如：String类，是要找ASCII大小比较的，那么我们就以自定义的类来存储TreeSet中，使其具备可比性。

package www.fuxi.jihe;
//自定义的异常
public class RunntimeNoStudentExceptionextends RuntimeException {
    publicRunntimeNoStudentException(String message){
      super(message);
    }
}
 
package www.fuxi.jihe;
 
public class Student implements Comparable {
   private String name;// 姓名
   private int age;// 年龄
  
   public Student(Stringname, int age) {
     super();
     this.name = name;
     this.age = age;
   }
 
   public String getName() {
     return name;
   }
 
   public int getAge() {
     return age;
   }
 
   public int compareTo(Object o) {
     if(!(o instanceof Student))
        throw new RunntimeNoStudentException("不是Student对象");
     Student stu=(Student)o;
     if(this.name.equals(stu.name)){
        return this.age-stu.age;
     }
     return this.name.compareTo(stu.name);
   }
 
}
package www.fuxi.jihe;
 
import java.util.Iterator;
import java.util.TreeSet;
 
public class TreeSetDemo {
   public static void main(String[]args) {
     TreeSet set=new TreeSet();
     set.add(new Student("zhangsan",22));
     set.add(new Student("zhangsan",23));
     set.add(new Student("zhangsan",22));
     set.add(new Student("lisi",25));
     Iterator it=set.iterator();
     while(it.hasNext()){
        Student stu=(Student)it.next();
        System.out.println(stu.getName()+":"+stu.getAge());
     }
   }
 
}
 
结果：
lisi:25
zhangsan:22
zhangsan:23
 

从结果可以看出，如果名字相同，则比较年龄，如果年龄也相同，那么对象就相同了，此不存入此对象。

3. 模拟二叉树

package www.fuxi.jihe;
 
class ErChaShu {
   class Node {
     private Comparable data;
     private Node left;
     private Node right;
 
     public Node(Comparable data) {// 构造方法初始化数据
        this.data = data;
     }
 
     public void addNode(Node newNode) {
        if (newNode.data.compareTo(this.data) < 0) {// 利用compareTo方法比较
          if (this.left == null) {// 判断是否放在左子树
             this.left = newNode;
          } else {
             this.left.addNode(newNode);
          }
        }
        if (newNode.data.compareTo(this.data) >= 0) {// 判断是否放在右子树
          if (this.right == null) {
             this.right = newNode;
          } else {
             this.right.addNode(newNode);
          }
        }
 
     }
 
     public void printNode() {// 输出元素，中序遍历
        if (this.left != null) {
          this.left.printNode();// 输出左子树
        }
        System.out.print(this.data + "\t");
        if (this.right != null) {
          this.right.printNode();
        }
     }
   }
 
   private Node root = null;// 存放根节点
 
   public void add(Comparable com) {// 加入新元素
     Node newNode = new Node(com);// 定义新结点
     if (this.root == null) {
        this.root = newNode;
     } else {
        this.root.addNode(newNode);// 判断是放在左子树还是右子树
     }
   }
 
   public void print() {// 输出新结点
     this.root.printNode();
   }
}
 
public class TextClass {
   public static void main(String[] agrs) throws Exception {
     ErChaShu t = new ErChaShu();
     t.add(4);// 添加元素
     t.add(2);
     t.add(0);
     t.add(7);
     t.add(9);
     t.add(8);
     t.print();// 输出元素
   }
}
 
结果：
0  2  4  7  8  9 

4. 自定义比较器（Comparator）

除了第一种比较方式，自然的排序，是对象本身具备比较性，还有另一种比较方式，那就是使集合具备比较性，那就是自定义比较器，把比较器通过构造方式传给集合，使集合剧本比较性。

步骤：定义一个类，实现接口Comparator，然后重写方法compare方法。

当比较器和自定义的自然排序同时存在的时候，那么就一比较器为主。

public class RunntimeNoStudentException extends RuntimeException {
    publicRunntimeNoStudentException(String message){
      super(message);
    }
}
public class Person {
   private String name;// 姓名
   private int age;// 年龄
 
   public Person(String name, int age) {
     super();
     this.name = name;
     this.age = age;
   }
 
   public String getName() {
     return name;
   }
 
   public int getAge() {
     return age;
   }
 
  
}
import java.util.Comparator;
 
public class MyCompartor implements Comparator {
 
   public int compare(Object o1, Object o2) {//从写此方法
     if(!(o1 instanceof Person || o2 instanceof Person)){
         throw newRunntimeNoStudentException("不是Person对象");
     }
     Person p=(Person)o1;
     Person p1=(Person)o2;
     int num=p.getName().compareTo(p1.getName());
     if(num==0)
        return p.getAge()-p1.getAge();
     return num;
   }
 
}
package www.fuxi.jihe;
 
import java.util.Iterator;
import java.util.TreeSet;
 
public class TreeSetDemo {
   publicstatic void main(String[] args) {
    
     TreeSetset = new TreeSet(new MyCompartor());
     set.add(newPerson("java03", 22));
     set.add(newPerson("java01", 23));
     set.add(newPerson("net07", 22));
     set.add(newPerson("net03", 25));
     Iteratorit = set.iterator();
     while(it.hasNext()) {
        Personstu = (Person) it.next();
        System.out.println(stu.getName()+ ":" + stu.getAge());
     }
   }
 
}
结果：
java01:23
java03:22
net03:25
net07:22
 

从结果可以看出，先是按照名字排序，然后按照年龄排序。

5. 练习

自定义比较器，存储字符串，字符串按照长度大小排序

import java.util.Comparator;
 
public class MyStringCompartor implements Comparator {
 
  public int compare(Object o1, Object o2) {//从写此方法
    if(!(o1 instanceof String|| o2 instanceof String)){
       throw newRunntimeNoStudentException("不是String的子类或者Person对象");
    }
    Strings1=(String)o1;
    Strings2=(String)o2;
    return s1.length()-s2.length();
  }
 
}
importjava.util.Iterator;
importjava.util.TreeSet;
 
publicclass TreeSetDemo {
public static void main(String[] args) {
  
   TreeSet set = new TreeSet(new MyStringCompartor());
   set.add("abcd");
   set.add("ad");
   set.add("cdf");
   set.add("a");
   set.add("abcdfrr");
   Iterator it = set.iterator();
   while (it.hasNext()) {
     System.out.println(it.next());
   }
}
 
}
结果：
a
ad
cdf
abcd
abcdfrr
 
 

--------------------ASP.Net+Android+IOS开发、.Net培训、期待与您交流！ --------------------