[Java] 학생 리스트 정렬 (List, Comparable, Comparator)

 

  학번, 이름, 국어점수, 영어점수, 수학점수, 총점, 등수를 멤버로 갖는
  Student클래스를 만든다.
  생성자는 학번, 이름, 국어, 영어, 수학 점수만 매개변수로 받아서 처리한다.
  
  이 Student객체들은 List에 저장하여 관리한다.
  List에 저장된 데이터들을 학번의 오름차순으로 정렬하여 출력하는 부분과
  총점의 역순으로 정렬하는 부분을 프로그램 하시오.
  (총점이 같으면 학번의 내림차순으로 정렬되도록 한다.)
  (학번 정렬기준은 Student클래스 자체에서 제공하도록 하고,
   총점 정렬기준은 외부클래스에서 제공하도록 한다.)

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

public class Homework01 {
	public static void main(String[] args) {

		//학번 스트링으로 
		
		List<Student> stuList = new ArrayList<Student>();
		stuList.add(new Student("1", "홍길동", 70, 80, 90));
		stuList.add(new Student("2", "변학도", 60, 100, 50));
		stuList.add(new Student("5", "강감찬", 60, 40, 30));
		stuList.add(new Student("8", "성춘향", 70, 80, 90));
		stuList.add(new Student("9", "일지매", 40, 30, 70));
		
		for (Student stu : stuList) {
            int rank = 1;
            for (Student stu2 : stuList) {
                if (stu2.getSum() > stu.getSum()) {
                    rank++;
                }
            }
            stu.setRank(rank);
        }
		
		Collections.shuffle(stuList);
		System.out.println("섞은 후: ");
		for (Student stu : stuList) {
			System.out.println(stu);
		}
		
		
		
		Collections.sort(stuList);
		System.out.println("학번 오름차순 정렬: ");
		for (Student student : stuList) {
			System.out.println(student);
		}
		
		
		System.out.println("총점 내림차순(총점이 같으면 학번 내림차순): ");
		Collections.sort(stuList,new Desc());
		for (Student stu : stuList) {
			System.out.println(stu);
		}

	}

}

class Student implements Comparable<Student>{
	private String stuNo;
	private String name;
	private int kor;
	private int eng;	
	private int math;
	private int sum;
	private int rank;
	
	public Student(String stuNo, String name, int kor, int eng, int math) {
		super();
		this.stuNo = stuNo;
		this.name = name;
		this.kor = kor;
		this.eng = eng;
		this.math = math;
		this.sum = kor+eng+math;
	}

	@Override
	public String toString() {
		return "Student [stuNo=" + stuNo + ", name=" + name + ", kor=" + kor + ", eng=" + eng + ", math=" + math
				+ ", sum=" + sum + ", rank=" + rank + "]";
	}

	public String getStuNo() {
		return stuNo;
	}

	public void setStuNo(String stuNo) {
		this.stuNo = stuNo;
	}

	public String getName() {
		return name;
	}

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

	public int getKor() {
		return kor;
	}

	public void setKor(int kor) {
		this.kor = kor;
	}

	public int getEng() {
		return eng;
	}

	public void setEng(int eng) {
		this.eng = eng;
	}

	public int getMath() {
		return math;
	}

	public void setMath(int math) {
		this.math = math;
	}

	public int getSum() {
		return sum;
	}

	public void setSum(int sum) {
		this.sum = sum;
	}

	public int getRank() {
		return rank;
	}

	public void setRank(int rank) {
		this.rank = rank;
	}
	
	//학번을 기준으로 오름차순 정렬하기
	@Override
	public int compareTo(Student stu) {
		
		return this.getStuNo().compareTo(stu.getStuNo());
	}
	
}


//외부정렬자 comparator 구현 (총점이 같으면 학번의 내림차순으로 정렬)
class Desc implements Comparator<Student> {
	
	
	@Override
	public int compare(Student stu1, Student stu2) {
		// 총점 기준 내림차순으로 정렬
		int res = new Integer(stu1.getSum()).compareTo(stu2.getSum())*-1;
		// 총점이 같으면 학번을 내림차순으로 정렬
		if(res == 0)  {
			return stu1.getStuNo().compareTo(stu2.getStuNo())*-1;
		}
		return res;
	}
}