[백준] 10814번. 나이순 정렬 (Java)

[문제 바로가기] https://www.acmicpc.net/problem/10814

문제

온라인 저지에 가입한 사람들의 나이와 이름이 가입한 순서대로 주어진다. 이때, 회원들을 나이가 증가하는 순으로, 나이가 같으면 먼저 가입한 사람이 앞에 오는 순서로 정렬하는 프로그램을 작성하시오.

입력

첫째 줄에 온라인 저지 회원의 수 N이 주어진다. (1 ≤ N ≤ 100,000)

둘째 줄부터 N개의 줄에는 각 회원의 나이와 이름이 공백으로 구분되어 주어진다. 나이는 1보다 크거나 같으며, 200보다 작거나 같은 정수이고, 이름은 알파벳 대소문자로 이루어져 있고, 길이가 100보다 작거나 같은 문자열이다. 입력은 가입한 순서로 주어진다.

출력

첫째 줄부터 총 N개의 줄에 걸쳐 온라인 저지 회원을 나이 순, 나이가 같으면 가입한 순으로 한 줄에 한 명씩 나이와 이름을 공백으로 구분해 출력한다.

풀이

1. 2가지 방법으로 객체를 생성해보자!
   1. 공백을 두고 나이와 이름을 받으므로 String[][]형태로 객체 생성
   2. C언어에서 Node와 같이 Person 클래스 객체 생성해서 해보기
2. 나이는 Integer.parseInt() 사용해서 값 비교
3. 이번에는 Arrays.sort()를 Comparator<>() 이용해서 오버라이드 해보기!
4. compare()은 값이 같으면 객체 위치를 바꾸지 않기 때문에 자동으로 입력 순으로 정렬된다.

1-1방법의 경우 String의 배열 크기를 N x 2 로 하지 못하고 N x N 해서 메모리 초과가 발생했다.

배열을 생성할 때는 배열의 크기를 꼭 다시한번 확인하자!

코드

1-1. String 타입으로 2차원 배열만들기

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		String[][] str = new String[N][2];
		
		for(int i = 0; i < N; i++) {
			str[i][0] = sc.next();
			str[i][1] = sc.next();
		}
		
		//나이 순 -> 같으면 가입 순
		Arrays.sort(str, new Comparator<String[]>() {
			//나이 순으로 정렬
			//나이가 같으면 입력된 그대로 둠 -> 자동으로 입력 순으로 정렬
			@Override	
			public int compare(String[] s1, String[] s2) {
				return Integer.parseInt(s1[0]) - Integer.parseInt(s2[0]);
			}
		});
		
		for(int i = 0; i < N; i++) {
			System.out.println(str[i][0] + " " + str[i][1]);
		}
	}
}

1-2. Person 클래스 객체 생성하여 배열 만들기

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;

public class Main {
	
	public static class Person{
		
		private int age;
		private String name;
		
		public Person() {};
		
		public Person(int age, String name) {
			this.age = age;
			this.name = name;
		}

		public int getAge() {
			return age;
		}

		public String getName() {
			return name;
		}
		
		@Override
		public String toString() {
			return age + " " + name;
		}
	}
	
	public static void main(String[] args) throws Exception {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		Person[] p = new Person[N];
		
		for(int i = 0; i < N; i++) {
			p[i] = new Person(sc.nextInt(), sc.next());
		}
		
		Arrays.sort(p, new Comparator<Person>() {
			//나이 순으로 정렬
			//나이가 같으면 입력된 그대로 둠 -> 자동으로 입력 순으로 정렬
			@Override	
			public int compare(Person p1, Person p2) {
				return p1.getAge() - p2.getAge();
			}
		});
		
		for(int i = 0; i < N; i++) {
			System.out.println(p[i]);
		}
	}
}