[정렬] 10814번 - 나이순 정렬

🔎 <문제> 나이순 정렬

• 주의할 점
  나이순 정렬을 기준으로 하되, 나이가 같을 경우 주어진 입력순으로 정렬한다.

  이때 정렬방법은 Arrays.sort()에 Comparator의 compare 메소드를 구현하여 사용자 정렬을 사용해서 쓸 수 있다. 즉, 나이순으로 정렬하면서 이름은 따로 비교를 안한다면 나이순으로 정렬하되, 나이가 같을 경우는 0이 반환되어 자연스레 입력순으로 정렬된다.

📝 풀이 코드

Person[] + BufferedReader + StringBuilder를 사용하여 문제를 풀었다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;

public class Main {
	public static void main(String args[])throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int N = Integer.parseInt(br.readLine());
		Person[] personArray = new Person[N];
		
		for(int i = 0; i < N; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			int age = Integer.parseInt(st.nextToken());
			String name = st.nextToken();
			personArray[i] =new Person(age, name);
		}
		
		Arrays.sort(personArray, new Comparator<Person>() {
			public int compare(Person p1, Person p2) {
				return p1.age - p2.age;
			}
		});
		
		StringBuilder sb = new StringBuilder();
		
		for(int i = 0; i < N; i++) {
        	// 객체배열의 객체를 출력하면 해당 인덱스의 객체의 toString() 이 출력 됨
			sb.append(personArray[i]);
		}
		
		System.out.println(sb);
	}
	
	public static class Person{
		int age;
		String name;
		
		public Person(int age, String name) {
			this.age = age;
			this.name = name;
		}
		
        @Override
		public String toString() {
			return age + " " + name + "\n";
		}
		
	}
}

📍toString() 메소드

객체를 출력할 때, 사용자가 임의로 출력하고자 하는 문자열을 지정할 수 있다.

---

처음으로 머릿속에서 문제 접근방향과 코드가 머릿속에서 술술 나와서 막힘없이 문제를 풀 수 있었다!

Reference

[백준] 10814번 : 나이순 정렬
st님 티스토리