Java中Treeset自定义排序的错误

Question

在书上找的一串Treeset自定义排序代码，为什么最后没能正确输出？

import java.util.Comparator;

import java.util.TreeSet;

class Student{
    private String name;
    private String id;/*age数据类型不能换为intCompareTo是String类的一个方法*/
    public Student(String id,String name){
        this.name=name;
        this.id=id;
    }
    public String getId(){
        return id;
    }
    public String getName(){
        return name;
    }

    public String toString() {
        return id+":"+name;
    }
}
public class Desdend {
    public static void main(String[] args){
        TreeSet ts = new TreeSet(new Comparator() {
            @Override
            public int compare(Object o1, Object o2) {
                Student stu1 = (Student) o1;
                Student stu2 = (Student) o2;
                if (!stu1.getId().equals(stu2.getId())) {
                    return stu1.getId().compareTo(stu2.getId());
                } else {
                    return stu1.getName().compareTo(stu2.getName());
                }
            }
        });
        ts.add(new Student("38","张三"));
        ts.add(new Student("28","李四"));
        ts.add(new Student("78","王五"));
        ts.add(new Student("8","赵六"));
        System.out.println(ts);
    }
}

以上代码为什么没能对数据进行正确排序，与第27行有什么关系，为什么会是灰色

Answer 1

【以下回答由 GPT 生成】

问题描述：

我在书上找到了以下一串TreeSet自定义排序的Java代码，但是无法正确输出结果。我想要对学生名称和ID进行排序。这个代码有什么错误吗？能帮我进行优化吗？

import java.util.TreeSet;

class Student implements Comparable<Student> {
    private String name;
    private int id;

    public Student(String name, int id) {
        this.name = name;
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public int getId() {
        return id;
    }

    @Override
    public int compareTo(Student student) {
        return this.name.compareTo(student.name);
    }
}

public class Main {
    public static void main(String[] args) {
        TreeSet<Student> students = new TreeSet<>();
        students.add(new Student("Alice", 2));
        students.add(new Student("Bob", 1));
        students.add(new Student("Charlie", 3));

        for (Student student : students) {
            System.out.println(student.getName() + " - " + student.getId());
        }
    }
}

请检查代码中的compareTo方法是否可以正确比较学生的ID。另外，对名称和ID都进行排序，您可以使用Comparator来自定义排序逻辑。

以下是修复后的代码：

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

class Student implements Comparable<Student> {
    private String name;
    private int id;

    public Student(String name, int id) {
        this.name = name;
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public int getId() {
        return id;
    }

    @Override
    public int compareTo(Student student) {
        // 先按照姓名排序
        int nameComparison = this.name.compareTo(student.name);
        // 如果姓名相同，则按照ID排序
        if (nameComparison == 0) {
            return Integer.compare(this.id, student.id);
        }
        return nameComparison;
    }

    public static class StudentComparator implements Comparator<Student> {
        @Override
        public int compare(Student student1, Student student2) {
            // 先按照姓名排序
            int nameComparison = student1.name.compareTo(student2.name);
            // 如果姓名相同，则按照ID排序
            if (nameComparison == 0) {
                return Integer.compare(student1.id, student2.id);
            }
            return nameComparison;
        }
    }
}

public class Main {
    public static void main(String[] args) {
        TreeSet<Student> students = new TreeSet<>(new Student.StudentComparator());
        students.add(new Student("Alice", 2));
        students.add(new Student("Bob", 1));
        students.add(new Student("Charlie", 3));

        for (Student student : students) {
            System.out.println(student.getName() + " - " + student.getId());
        }
    }
}

这样修改后的代码会先按照姓名进行排序，如果姓名相同，则按照ID排序。

【相关推荐】

• 你可以参考下这个问题的回答, 看看是否对你有帮助, 链接: https://ask.csdn.net/questions/362348
• 我还给你找了一篇非常好的博客，你可以看看是否有帮助，链接：java中如果想要去重且排序用TreeSet
• 您还可以看一下 屈博老师的Java基础课程中的 7--TreeSet存放自定义对象小节, 巩固相关知识点
  
• 除此之外, 这篇博客: Java中TreeSet集合存储自定义类型中的 TreeSet集合存储自定义类型： 部分也许能够解决你的问题。

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^