백준 문제풀이

[Baekjoon 10825] 국영수 - JAVA

planting grass 2023. 6. 3. 18:57
728x90

[Silver IV] 국영수 - 10825

문제 링크

성능 요약

메모리: 62256 KB, 시간: 720 ms

분류

정렬

문제 설명

도현이네 반 학생 N명의 이름과 국어, 영어, 수학 점수가 주어진다. 이때, 다음과 같은 조건으로 학생의 성적을 정렬하는 프로그램을 작성하시오.

  1. 국어 점수가 감소하는 순서로
  2. 국어 점수가 같으면 영어 점수가 증가하는 순서로
  3. 국어 점수와 영어 점수가 같으면 수학 점수가 감소하는 순서로
  4. 모든 점수가 같으면 이름이 사전 순으로 증가하는 순서로 (단, 아스키 코드에서 대문자는 소문자보다 작으므로 사전순으로 앞에 온다.)

입력

첫째 줄에 도현이네 반의 학생의 수 N (1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 한 줄에 하나씩 각 학생의 이름, 국어, 영어, 수학 점수가 공백으로 구분해 주어진다. 점수는 1보다 크거나 같고, 100보다 작거나 같은 자연수이다. 이름은 알파벳 대소문자로 이루어진 문자열이고, 길이는 10자리를 넘지 않는다.

출력

문제에 나와있는 정렬 기준으로 정렬한 후 첫째 줄부터 N개의 줄에 걸쳐 각 학생의 이름을 출력한다.

문제 풀이

문제의 요구대로 정렬기준을 오버라이드 해주면 된다.

즉, Comparable 사용하여, 앞-뒤 계산 후 결과 값이 양수면 오름차순정렬, 음수면 내림차순정렬을 해주면 된다.

Comparable이 뭔지 자세하게 알고 싶다면 아래 주소에 관련 내용을 작성해뒀습니다.

2023.06.02 - [자바] - [JAVA] Comparable & Comparator

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

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

public class Main {

    static class Student {
        String name;
        int kor;
        int eng;
        int math;

        public Student(String name, int kor, int eng, int math) {
            this.name = name;
            this.kor = kor;
            this.eng = eng;
            this.math = math;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        ArrayList<Student> studentList = new ArrayList<>();
        int n = Integer.parseInt(br.readLine());

        while (n-- > 0) {
            st = new StringTokenizer(br.readLine());
            String name = st.nextToken();
            int kor = Integer.parseInt(st.nextToken());
            int eng = Integer.parseInt(st.nextToken());
            int math = Integer.parseInt(st.nextToken());

            studentList.add(new Student(name, kor, eng, math));
        }

        Collections.sort(studentList, new Comparator<Student>() {
            @Override
            public int compare(Student s1, Student s2) {

                if (s1.kor == s2.kor) {
                    if (s1.eng == s2.eng) {
                        if (s1.math == s2.math) {
                            return s1.name.compareTo(s2.name);
                        }
                        return s2.math - s1.math;
                    }
                    return s1.eng - s2.eng;
                }
                return s2.kor - s1.kor;
            }
        });

        StringBuilder sb = new StringBuilder();
        for (Student student : studentList) {
            sb.append(student.name).append('\n');
        }

        System.out.println(sb);
    }
}

코드 풀이

    static class Student {
        String name;
        int kor;
        int eng;
        int math;
  • 학생 클래스를 name, kor, eng, math 이름으로 정의한다.
        public Student(String name, int kor, int eng, int math) {
            this.name = name;
            this.kor = kor;
            this.eng = eng;
            this.math = math;
        }
    }

생성자를 만들어준다.

        ArrayList<Student> studentList = new ArrayList<>();
        int n = Integer.parseInt(br.readLine());
  • 학생 목록을 저장할 ArrayList 생성한다.
  • 학생 수 입력 받을 n을 생성한다.
        while (n-- > 0) {
            st = new StringTokenizer(br.readLine());
            String name = st.nextToken();
            int kor = Integer.parseInt(st.nextToken());
            int eng = Integer.parseInt(st.nextToken());
            int math = Integer.parseInt(st.nextToken());

            studentList.add(new Student(name, kor, eng, math));
        }
  • 입력된 학생 수만큼 반복하여 학생 정보 입력해준다.
  • 공백으로 구분된 학생 정보 입력해준다.
  • name, eng, math를 입력해준다.
  • 학생 객체를 생성하여 목록에 추가해준다.
        Collections.sort(studentList, new Comparator<Student>() {
            @Override
            public int compare(Student s1, Student s2) {

                if (s1.kor == s2.kor) {
                    if (s1.eng == s2.eng) {
                        if (s1.math == s2.math) {
                            return s1.name.compareTo(s2.name);
                        }
                        return s2.math - s1.math;
                    }
                    return s1.eng - s2.eng;
                }
                return s2.kor - s1.kor;
            }
        });
  • 학생 목록을 정렬하기 위해 Comparator를 사용하여 정렬 기준 지정해준다.
  • 국어 점수가 같을 경우, 영어 점수가 같을 경우, 수학 점수가 같을 경우 이름을 사전순으로 비교하여 오름차순 정렬해준다.
  • 국어 점수가 같을 경우, 영어 점수가 같을 경우, 수학 점수를 내림차순으로 정렬
  • 국어 점수가 같을 경우, 영어 점수를 오름차순으로 정렬
  • 국어 점수가 같이 않으면, 국어 점수를 내림차순으로 정렬
        StringBuilder sb = new StringBuilder();
        for (Student student : studentList) {
            sb.append(student.name).append('\n');
        }

        System.out.println(sb);
    }
}
  • 정렬된 학생 목록에서 이름을 하나씩 StringBuilder에 추가한다.
  • 결과 출력
728x90
저작자표시