黑马程序员_日记59_TreeMap自定义对象练习

　——- android培训、java培训、期待与您交流！ ———-

TreeMap自定义对象存取练习

需求：

每一个学生都有对应的归属地。
学生Student，地址String。
学生属性：姓名，年龄。
注意：姓名和年龄相同的视为同一个学生。
保证学生的唯一性。
对学生对象的年龄进行升序排序,用Comparable实现；
对学生的姓名进行字典序排序，用Comparator实现。

因为数据是以键值对形式存在的。
所以要使用可以排序的Map集合。TreeMap。

思路：

1 把学生属性存入学生对象，并抽象出学生类
2 创建TreeMap容器，并把学生对象存入
3 取出学生
4 创建比较器类并测试按姓名字典序排序

在写学生类要注意的地方：

1 实现Comparable接口，覆盖compareTo方法
2 去重，计算哈希值，覆盖hashCode和equals方法
3 覆盖toString方法
4 记得写全set ，get方法

源代码

import java.util.*;
//主类
public class TreeMapTest
{
    public static void sop(Object obj)
    {
        System.out.println(obj);
    }

    public static void main(String[] args)
    {
        //1 创建TreeMap映射对象
        //TreeMap<Student,String> tm = new TreeMap<Student,String>(new StuNameComparator());//姓名字典序排序
        TreeMap<Student,String> tm = new TreeMap<Student,String>();//年龄升序排序

        //2 添加Student对象
        tm.put(new Student("Jack",25),"北京");
        tm.put(new Student("Jack",22),"北京");
        tm.put(new Student("Mike",22),"天津");
        //sop(tm.put(new Student("Mike",22),"淮南"));//注意了，如果Student相同，但是地址不相同，根据put方法，新地址会覆盖原来的地址
        tm.put(new Student("Jimmy",21),"上海");
        tm.put(new Student("Jakie",20),"广州");
        tm.put(new Student("David",17),"长沙");
        tm.put(new Student("Tom",26),"连云港");

        //第一种方式取出所有键值
        sop("第一种方式取出所有键值");
         getWay(tm);

        //第二种方式取出所有键值
        //sop("第二种方式取出所有键值");
         //getWay_2(tm);
    }

    //第一种方式迭代取出所有键值
    public static void getWay(TreeMap<Student,String> tm)
    {
        //1 把所有key存入set
        Set<Student> keySet = tm.keySet();

        //2 获取迭代器
        for(Iterator<Student> it = keySet.iterator(); it.hasNext(); )
        {
            //3 迭代取出key
            Student stu = it.next();
            //4 通过key获取value
            String addr = tm.get(stu);

            //打印取出结果
            sop(stu.toString()+".........."+addr);
        }
    }

    //第二种方式迭代取出所有键值
    public static void getWay_2(TreeMap<Student,String> tm)
    {
        //1 把所有映射项存入Map.Entry
        Set<Map.Entry<Student,String>> entry = tm.entrySet();

        //2 获取迭代器
        for(Iterator<Map.Entry<Student,String>> it = entry.iterator(); it.hasNext(); )
        {
            //3 迭代取出映射项
            Map.Entry<Student,String> me = it.next();

            //4 利用Map.Entry中的get方法获取key和value
            Student stu = me.getKey();
            String addr = me.getValue();

            //5打印取出结果
            sop(stu.toString()+"***********"+addr);
        }
    }
}

//学生姓名比较器类
class StuNameComparator implements Comparator<Student>
{
    //覆盖compare方法
    public int compare(Student stu1,Student stu2)
    {
        //先按姓名排序，如果相同再按年龄升序排序
        int num = stu1.getName().compareTo(stu2.getName());

        if(num == 0)
            return new Integer(stu1.getAge()).compareTo(new Integer(stu2.getAge()));

        //否则就按按姓名排序
        return  num;
    }
}


//学生类
class Student implements Comparable<Student>
{
    private String name;
    private int age;

    //构造函数初始化学生
    Student(String name,int age)
    {
        this.name = name;
        this.age = age;
    }

    public int compareTo(Student stu)
    {
        //按学生的年龄排序，如果年龄相等就按姓名字典序排序
        int num = new Integer(this.getAge()).compareTo(new Integer(stu.getAge()));

        if(num == 0)
            return this.getName().compareTo(stu.getName());

        //否则就按年龄排序
        return num;
    }

    //计算学生对象的哈希地址
    public int hashCode()
    {
        return name.hashCode()+age*37;
    }

    public boolean equals(Object obj)
    {
        //如果不是学生对象，抛出类型转换异常
        if(!(obj instanceof Student))
            throw new ClassCastException("不能转换成Student类型");

        //类型转换
        Student stu = (Student)obj;

        //比较学生对象姓名和年龄，去除重复元素
        return this.getName().equals(stu.getName()) && this.getAge()==stu.getAge();
    }

    public void setName(String name)
    {
        this.name =name;
    }

    public String getName()
    {
        return name;
    }

    public void setAge(int age)
    {
        this.age = age;
    }

    public int getAge()
    {
        return age;
    }

    public String toString()
    {
        return name+"....."+age;
    }
}

注意

以下操作均在主类的主函数进行。

如果想要第一种或者第二种方式迭代取出所有键值，请打开或者关闭对于注释语句。

如果想按姓名字典序排序，请打开对应语句，并关闭按年龄升序排序语句；

如果想按年龄升序排序，请打开对应语句，并关闭姓名字典序排序。

当存入重复的学生，不同的地址时候，put方法会返回被覆盖的地址