1. 
   
2. import java.io.BufferedReader;
   
3. import java.io.BufferedWriter;
   
4. import java.io.FileWriter;
   
5. import java.io.IOException;
   
6. import java.io.InputStreamReader;
   
7. import java.util.Collections;
   
8. import java.util.Comparator;
   
9. import java.util.Set;
   
10. import java.util.TreeSet;
    
11. 
    
12. /*
    
13. * 把学生成绩存储在stu.txt文件中，并按照总分由高到低的顺序排序
    
14. */
    
15. 
    
16. public class StudentInfoTest {
    
17. 
    
18.         public static void main(String[] args) {
    
19.                 Comparator<Student> comp=Collections.reverseOrder();
    
20.                 Set<Student> set=StuInfoTool.getInfo(comp);
    
21.                 StuInfoTool.WriteInfoToTest(set);
    
22.         }
    
23. 
    
24. }
    
25. 
    
26. //定义工具类
    
27. class StuInfoTool{
    
28.        
    
29.         //通过键盘获取学生成绩
    
30.         public static Set<Student> getInfo(){
    
31.                 return getInfo(null);
    
32.         }
    
33.         public static Set<Student> getInfo(Comparator<Student> comp){                               
    
34.                 BufferedReader br=null;
    
35.                 Set<Student> set=null;
    
36.                 if(comp==null){
    
37.                         set=new TreeSet<Student>();
    
38.                 }else{
    
39.                         set=new TreeSet<Student>(comp);
    
40.                 }
    
41.                 try{
    
42.                         br=new BufferedReader(new InputStreamReader(System.in));
    
43.                         String line=null;
    
44.                         while((line=br.readLine())!=null){
    
45.                                 if(line.equals("over")){
    
46.                                         break;
    
47.                                 }
    
48.                                 String[] info=line.split(",");
    
49.                                 Student stu=new Student(info[0],Integer.parseInt(info[1]),Integer.parseInt(info[2]),Integer.parseInt(info[3]));
    
50.                                 set.add(stu);                       
    
51.                         }
    
52.                 }catch(IOException e){
    
53.                         throw new RuntimeException("输入失败");
    
54.                 }finally{
    
55.                         try {
    
56.                                 if (br != null) {
    
57.                                         br.close();
    
58.                                 }
    
59.                         } catch (IOException e2) {
    
60.                                 throw new RuntimeException("输入流关闭失败");
    
61.                         }
    
62.                 }
    
63.                 return set;
    
64.         }
    
65.        
    
66.         //将获取的信息保存到文本文件中
    
67.         public static void WriteInfoToTest(Set<Student> set){
    
68.                 BufferedWriter bw=null;
    
69.                 try{
    
70.                         bw=new BufferedWriter(new FileWriter("stuinfo.txt"));
    
71.                         for(Student stu:set){
    
72.                                 bw.write(stu.toString());
    
73.                                 bw.newLine();
    
74.                                 bw.flush();
    
75.                         }
    
76.                 }catch(IOException e){
    
77.                         throw new RuntimeException("写入文本文件失败");
    
78.                 }finally{
    
79.                         try {
    
80.                                 if (bw != null) {
    
81.                                         bw.close();
    
82.                                 }
    
83.                         } catch (IOException e2) {
    
84.                                 throw new RuntimeException("写入流关闭失败");
    
85.                         }
    
86.                 }
    
87.         }
    
88. }
    
89. 
    
90. //学生类
    
91. class Student implements Comparable<Student>{
    
92.         private String name;
    
93.         private int ch,ma,en,sum;       
    
94.        
    
95.         public Student(String name, int ch, int ma, int en) {
    
96.                 this.name = name;
    
97.                 this.ch = ch;
    
98.                 this.ma = ma;
    
99.                 this.en = en;               
    
100.                 sum=ch+ma+en;
     
101.         }
     
102.        
     
103.         @Override
     
104.         public int hashCode() {
     
105.                 return name.hashCode()+sum;
     
106.         }
     
107.        
     
108.         @Override
     
109.         public boolean equals(Object obj) {
     
110.                 if(!(obj instanceof Student)){
     
111.                         throw new RuntimeException("类型不匹配");
     
112.                 }
     
113.                 Student stu=(Student)obj;
     
114.                 return this.name.equals(stu.name) && this.ch==stu.ch && this.ma==stu.ma && this.en==stu.en;
     
115.         }
     
116. 
     
117.         @Override
     
118.         public int compareTo(Student stu) {
     
119.                 int num=new Integer(this.sum).compareTo(new Integer(stu.sum));
     
120.                 if(num==0){
     
121.                         return this.name.compareTo(stu.name);
     
122.                 }
     
123.                 return sum;
     
124.         }
     
125.        
     
126.         @Override
     
127.         public String toString() {
     
128.                 return name+":"+ch+" "+ma+" "+en+"\t"+sum;
     
129.         }
     
130. }
     

复制代码

treeset比较需要覆盖hashcode和equals方法？

陈果 发表于 2014-3-19 22:57
写错了，我太粗心了，应该是num

呵……同道中人，我也经常干这种事，都有经验了

方青木 发表于 2014-3-19 22:53
public int compareTo(Student stu) {

119.                int num=new Integer(this.sum).compareTo(new ...

写错了，我太粗心了，应该是num

平凡成就非凡 发表于 2014-3-19 22:50
123行  return sum;错啦

嗯，就是这个地方，改了以后就对了

public int compareTo(Student stu) {

119.                int num=new Integer(this.sum).compareTo(new Integer(stu.sum));

120.                if(num==0){

121.                        return this.name.compareTo(stu.name);

122.                }

123.                return sum;

124.        }
返回sum是啥意思啊

123行  return sum;错啦