美文网首页Java
Java 集合 TreeSet

Java 集合 TreeSet

作者: 一亩三分甜 | 来源:发表于2019-09-14 23:24 被阅读0次

TreeSet:可以对Set集合中的元素进行排序。底层数据结构是二叉树,保证元素唯一性的依据,compareTo方法return 0;

TreeSet排序的第一种方式:让元素自身具备比较性。元素需要实现Comparable接口,覆盖compareTo方法。这种方式也称为元素的自然顺序,或者叫做默认顺序。

TreeSet的第二种排序方式:让元素自身不具备比较性时。或者具备的比较性不是所需要的。这时就需要让集合自身具备比较性。在集合初始化时,就有了比较方式。

import java.util.Iterator;
import java.util.TreeSet;
public class TreeSetDemo {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet();
        ts.add("cba");
        ts.add("aaa");
        ts.add("bca");
        ts.add("Dbca");

        Iterator it = ts.iterator();
        while (it.hasNext())
        {
            System.out.println(it.next());
        }
    }
}
//输出
Dbca
aaa
bca
cba

需求:往TreeSet集合中存储自定义对象学生。

想按照学生的年龄进行排序。
记住:排序时,当主要条件相同时,一定判断一下次要条件。

import java.util.Iterator;
import java.util.TreeSet;

class Student0
{
    private String name;
    private int age;
    Student0(String name,int age)
    {
        this.name = name;
        this.age = age;
    }
    public String getName()
    {
        return name;
    }
    public int getAge()
    {
        return age;
    }
}
public class TreeSetDemo0 {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet();
        ts.add(new Student0("lisi02",22));
        ts.add(new Student0("lisi007",20));
        ts.add(new Student0("lisi09",19));
        ts.add(new Student0("lisi01",40));

        Iterator it = ts.iterator();
        while (it.hasNext())
        {
            Student0 stu = (Student0)it.next();
            System.out.println(stu.getName()+"..."+stu.getAge());
        }
    }
}
//输出
Exception in thread "main" java.lang.ClassCastException: Student0 cannot be cast to java.base/java.lang.Comparable
    at java.base/java.util.TreeMap.compare(TreeMap.java:1291)
    at java.base/java.util.TreeMap.put(TreeMap.java:536)
    at java.base/java.util.TreeSet.add(TreeSet.java:255)
    at TreeSetDemo0.main(TreeSetDemo0.java:25)

import java.util.Iterator;
import java.util.TreeSet;

class Student0 implements Comparable//该接口强制让学生具备比较性。
{
    private String name;
    private int age;
    Student0(String name,int age)
    {
        this.name = name;
        this.age = age;
    }
    public String getName()
    {
        return name;
    }
    public int getAge()
    {
        return age;
    }
    public int compareTo(Object obj)
    {
        if (!(obj instanceof Student0))
            throw new RuntimeException("不是学生对象");
        Student0 s = (Student0)obj;
        System.out.println(this.name + "...compareto....."+s.name);
        if (this.age > s.age)
            return 1;
        if (this.age == s.age)
            return (this.name.compareTo(s.name));
        return -1;
    }
}
public class TreeSetDemo0 {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet();
        ts.add(new Student0("lisi02",22));
        ts.add(new Student0("lisi007",20));
        ts.add(new Student0("lisi09",19));
        ts.add(new Student0("lisi01",40));
        ts.add(new Student0("lisi06",19));
        ts.add(new Student0("lisi01",40));

        Iterator it = ts.iterator();
        while (it.hasNext())
        {
            Student0 stu = (Student0)it.next();
            System.out.println(stu.getName()+"..."+stu.getAge());
        }
    }
}
//输出
lisi02...compareto.....lisi02
lisi007...compareto.....lisi02
lisi09...compareto.....lisi02
lisi09...compareto.....lisi007
lisi01...compareto.....lisi007
lisi01...compareto.....lisi02
lisi06...compareto.....lisi007
lisi06...compareto.....lisi09
lisi01...compareto.....lisi007
lisi01...compareto.....lisi02
lisi01...compareto.....lisi01
lisi06...19
lisi09...19
lisi007...20
lisi02...22
lisi01...40

当元素自身不具备比较性,或者具备的比较性不是所需要的。这时需要让容器自身具备比较性。定义了比较器,将比较器对象作为参数传递给TreeSet集合的构造函数。
当两种排序都存在时,以比较器为主。定义一个类,实现Compatator接口,覆盖compare方法。

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

class Student1
{
    private String name;
    private int age;
    Student1(String name,int age)
    {
        this.name = name;
        this.age = age;
    }
    public String getName()
    {
        return name;
    }
    public int getAge()
    {
        return age;
    }
}
public class TreeSetDemo1 {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet(new MyCompare());
        ts.add(new Student1("lisi02",22));
        ts.add(new Student1("lisi007",20));
        ts.add(new Student1("lisi09",19));
        ts.add(new Student1("lisi01",40));
        ts.add(new Student1("lisi06",19));
        ts.add(new Student1("lisi01",40));

        Iterator it = ts.iterator();
        while (it.hasNext())
        {
            Student1 stu = (Student1)it.next();
            System.out.println(stu.getName()+"..."+stu.getAge());
        }
    }
}
class MyCompare implements Comparator
{
    public int compare(Object o1,Object o2)
    {
        Student1 s1 = (Student1) o1;
        Student1 s2 = (Student1) o2;
        int num = s1.getName().compareTo(s2.getName());
        if (num == 0)
        {
            if (s1.getAge()> s2.getAge())
                return 1;
            if (s1.getAge() == s2.getAge())
                return 0;
            return -1;
        }
        return num;
    }
}
//输出
lisi007...20
lisi01...40
lisi02...22
lisi06...19
lisi09...19

调用Integer对象的compareTo方法进行比较

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

class Student2
{
    private String name;
    private int age;
    Student2(String name,int age)
    {
        this.name = name;
        this.age = age;
    }
    public String getName()
    {
        return name;
    }
    public int getAge()
    {
        return age;
    }
}
public class TreeSetDemo2 {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet(new MyCompare0());
        ts.add(new Student2("lisi02",22));
        ts.add(new Student2("lisi007",20));
        ts.add(new Student2("lisi09",19));
        ts.add(new Student2("lisi01",40));
        ts.add(new Student2("lisi06",19));
        ts.add(new Student2("lisi01",40));
        ts.add(new Student2("lisi01",39));
        Iterator it = ts.iterator();
        while (it.hasNext())
        {
            Student2 stu = (Student2)it.next();
            System.out.println(stu.getName()+"..."+stu.getAge());
        }
    }
}
class MyCompare0 implements Comparator
{
    public int compare(Object o1,Object o2)
    {
        Student2 s1 = (Student2) o1;
        Student2 s2 = (Student2) o2;
        int num = s1.getName().compareTo(s2.getName());
        if (num == 0)
        {
            return new Integer(s1.getAge()).compareTo(new Integer(s2.getAge()));
        }
        return num;
    }
}
//输出
lisi007...20
lisi01...39
lisi01...40
lisi02...22
lisi06...19
lisi09...19

按照字符串长度排序

字符串本身具备比较性,但是它的比较方式不是所需要的。
这时就只能使用比较器。

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetDemo3 {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet(new MyCompare1());
        ts.add("abcd");
        ts.add("cc");
        ts.add("cba");
        ts.add("aaa");
        ts.add("z");
        ts.add("hahaha");
        Iterator it = ts.iterator();
        while (it.hasNext())
        {
            System.out.println(it.next());
        }
    }
}
class MyCompare1 implements Comparator
{
    public int compare(Object o1,Object o2)
    {
        String s1 = (String)o1;
        String s2 = (String)o2;
        int num = new Integer(s1.length()).compareTo(new Integer(s2.length()));
        if (num == 0)
        {
            return s1.compareTo(s2);
        }
        return num;
    }
}
//输出
z
cc
aaa
cba
abcd
hahaha

匿名内部类写法

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetDemo4 {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet(new Comparator() {
            @Override
            public int compare(Object o1, Object o2) {
                String s1 = (String)o1;
                String s2 = (String)o2;
                int num = new Integer(s1.length()).compareTo(new Integer(s2.length()));
                if (num == 0)
                {
                    return s1.compareTo(s2);
                }
                return num;
            }
        });
        ts.add("abcd");
        ts.add("cc");
        ts.add("cba");
        ts.add("aaa");
        ts.add("z");
        ts.add("hahaha");
        Iterator it = ts.iterator();
        while (it.hasNext())
        {
            System.out.println(it.next());
        }
    }
}
//输出
z
cc
aaa
cba
abcd
hahaha

倒序排列

import java.util.Comparator;
import java.util.Iterator;
import java.util.TreeSet;

public class TreeSetDemo5 {
    public static void main(String[] args) {
        TreeSet ts = new TreeSet(new MyCompare2());
        ts.add("abcd");
        ts.add("cc");
        ts.add("cba");
        ts.add("aaa");
        ts.add("z");
        ts.add("hahaha");
        Iterator it = ts.iterator();
        while (it.hasNext())
        {
            System.out.println(it.next());
        }
    }
}
class MyCompare2 implements Comparator
{
    public int compare(Object o1,Object o2)
    {
        String s1 = (String)o1;
        String s2 = (String)o2;
        System.out.println(s1+"...compareTo..."+s2);
        int num = new Integer(s2.length()).compareTo(new Integer(s1.length()));
        if (num == 0)
        {
            return s2.compareTo(s1);
        }
        return num;
    }
}
//输出
abcd...compareTo...abcd
cc...compareTo...abcd
cba...compareTo...abcd
cba...compareTo...cc
aaa...compareTo...cba
aaa...compareTo...cc
z...compareTo...cba
z...compareTo...cc
hahaha...compareTo...cba
hahaha...compareTo...abcd
hahaha
abcd
cba
aaa
cc
z

疑问点:TreeSet的比较次数发生了变化,第一次出现了自己和自己比较。每次比较也不是从集合的第一个元素abcd开始。

相关文章

网友评论

    本文标题:Java 集合 TreeSet

    本文链接:https://www.haomeiwen.com/subject/svczyctx.html