package TreeSetTest; import java.util.Iterator; import java.util.TreeSet; import javax.management.RuntimeErrorException; /* 可以对set集合中的元素进行排序,其底层的数据结构是二叉树, 保证元素唯一性的依据是compareTo和return 0; TreeSet排序的第一种方式 让元素自身具备比较性 元素需要实现Comparable接口,覆盖compareTo方法 这种方式也成为元素的自然顺序,或者叫做默认顺序 TreeSet排序的第二种方式 当元素自身不具备比较性时,或者具备的比较性不是所需的 这时就需要让集合自身具备比较性 在集合初始化时,就有了比较方式 记住:当比较时,如果主要条件相同,就比较次要条件 */ class Student implements Comparable // 该接口强制让对象具有比较性 { private String name; private int age; Student(String name,int age) { this.name = name; this.age = age; } public int compareTo(Object obj) { /* if (!(obj instanceof Student)) { throw new RuntimeException("不是学生对象"); } Student s = (Student)obj; // System.out.println(this.name+"=====Compare====="+this.age); if (this.age >s.age) { return 1; } else if (this.age == s.age) { // 如果年龄相同比较姓名 return this.name.compareTo(s.name); } else { return -1; } */ return -1; } public String getname() { return this.name; } public int getage() { return age; } } public class TreeSetDemo { public static void main(String[] args) { TreeSet<Student> tsSet =new TreeSet<Student>(); tsSet.add(new Student("as", 12)); tsSet.add(new Student("fvhgj",45)); tsSet.add(new Student("ert", 15)); tsSet.add(new Student("wenjia", 31)); tsSet.add(new Student("wejia", 1)); Iterator<Student> iterator = tsSet.iterator(); while (iterator.hasNext()) { Student stu = (Student)iterator.next(); System.out.println(stu.getage()+"======"+stu.getname()); } } } //////////////////////////////////////////// //////////////////////////////////////////// package TreeSetTest; import java.util.Comparator; import java.util.Iterator; import java.util.TreeSet; /* 当元素自身不具备比较性,或者具备的比较性不是所需要的。 这时需要让容器自身具备比较性。 定义了比较器,将比较器对象作为参数传递给TreeSet集合的构造函数。 当两种排序都存在时,一比较器为主 定义一个类,实现Comparator接口,覆盖compare方法 二叉树都是以 return 0;判断元素是否相等 */ public class TreeSetDemoTwo { public static void main(String[] args) { TreeSet<Student> tsSet = new TreeSet<Student>(new MyCompare()); tsSet.add(new Student("wenjai10", 29)); tsSet.add(new Student("wenjai11", 28)); tsSet.add(new Student("wenjai12", 27)); tsSet.add(new Student("wenjai13", 26)); tsSet.add(new Student("wenjai11", 21)); Iterator<Student> it = tsSet.iterator(); while (it.hasNext()) { Student student = (Student) it.next(); System.out.println(student.getname()+"======"+student.getage()); } } } class MyCompare implements Comparator { public int compare(Object o1,Object o2) { Student s1 = (Student)o1; Student s2 = (Student)o2; int num = s1.getname().compareTo(s2.getname()); if (num == 0) { // 由于整数类型也有比较方法,所以将整数封装成对象 return new Integer(s1.getage()).compareTo(new Integer(s2.getage())); /* if (s1.getage()>s2.getage()) { return 1; } if (s1.getage() == s2.getage()) { return 0; } return -1; */ } return num; } } /////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////// package TreeSetTest; import java.util.Comparator; import java.util.Iterator; import java.util.TreeSet; /* 练习: 按照字符串长度排序 字符串本身具有比较性,但是它的比较方式不是所需的 这时就只能用比较器 */ public class TreeSetPractice_One { public static void main(String[] args) { TreeSet<String> ts = new TreeSet<String>(new stringLenCompare()); ts.add("asd"); ts.add("bb"); ts.add("ba"); ts.add("ityior"); ts.add("hysdhbgg"); Iterator<String> it = ts.iterator(); for (String string : ts) { System.out.println(string); } } } class stringLenCompare implements Comparator { @Override public int compare(Object o1, Object o2) { // TODO Auto-generated method stub String str1 = (String)o1; String str2 = (String)o2; /* if (str1.length()>str2.length()) { return 1; } if(str1.length() == str2.length()) { return 0; } */ int num = new Integer(str1.length()).compareTo(new Integer(str2.length())); if (num == 0) { return str1.compareTo(str2); } return num; // return 0; } }
时间: 2024-10-03 22:51:27