HomeAbout Me

BOJ 10814 나이순 정렬

By kuper0201
Published in 알고리즘
2022-03-01
1 min read
BOJ 10814 나이순 정렬

Table Of Contents

01
문제
02
입력
03
풀이

문제

BOJ 10814 나이순 정렬 바로 가기

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


입력

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


풀이

회원들의 나이와 가입일을 기준으로 정렬하는 문제이다.
정렬 알고리즘은 안정 정렬불안정 정렬로 나눠진다.

안정 정렬(Stable Sort) : 정렬 기준이 같을 때 입력된 순서를 고려하여 정렬
불안정 정렬(Unstable Sort) : 입력된 순서를 고려하지 않고 정렬

각 정렬 알고리즘 중 대표적인 것은 다음과 같다.

안정 정렬(Stable Sort)불안정 정렬(Unstable Sort)
삽입 정렬(Insertion Sort)퀵 정렬(Quick Sort)
병합 정렬(Merge Sort)선택 정렬(Selection Sort)
버블 정렬(Bubble Sort)힙 정렬(Heap Sort)

각 언어가 사용하는 정렬 알고리즘은 다르므로 사용하는 언어에 따라 구현 방식이 다를수 있다. 각 언어별 구현은 아래에 설명해 두었다.

코드 보기(Java)
import java.io.*;
import java.util.*;

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());
        ArrayList<Member> members = new ArrayList<>();

        for(int i = 0; i < n; i++) {
            String member = br.readLine();

            int age = Integer.parseInt(member.split(" ")[0]);
            String name = member.split(" ")[1];

            members.add(new Member(age, name));
        }

        Collections.sort(members);

        for(int i = 0; i < members.size(); i++)
            System.out.println(members.get(i).age +" " +members.get(i).name);
    }
}

class Member implements Comparable<Member> {
    int age;
    String name;

    public Member(int age, String name) {
        this.age = age;
        this.name = name;
    }

    @Override
    public int compareTo(Member o) {
        if(this.age != o.age) return this.age - o.age;
        else return 0;
    }
}

Java Docs를 확인해 보면 자바의 Collections.sort() 함수는 안정 정렬인 팀 정렬(Tim Sort)을 사용한다.
따라서 나이가 다를 때만 비교값을 반환해주면 정렬이 완료된다.

코드 보기(C++)
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>

using namespace std;

bool cmp(pair<int, string> p1, pair<int, string> p2) {
    return p1.first < p2.first;
}

int main(int argc, const char * argv[]) {
    int n; cin >> n;
    
    vector<pair<int, string>> members;
    pair<int, string> member;
    
    for(int i = 0; i < n; i++) {
        cin >> member.first >> member.second;
        members.push_back(member);
    }
    
    stable_sort(members.begin(), members.end(), cmp);
    
    for(int i = 0; i < n; i++) {
        cout << members[i].first << " " << members[i].second << '\n';
    }
}

C++의 기본 sort() 함수는 불안정 정렬인 퀵 정렬(Quick Sort)을 사용하고 있다.
하지만 안정 정렬인 병합 정렬(Merge Sort)을 사용한 stable_sort() 함수를 따로 제공해 준다.
stable_sort() 함수를 이용하여 정렬을 완료한다.

코드 보기(Kotlin)
class Member {
    var age: Int
    var name: String

    constructor(age: Int, name:String) {
        this.age = age
        this.name = name
    }
}

fun main(args: Array<String>) {
    var N = readLine()!!.toInt()
    var members = ArrayList<Member>()

    for(i in 0 until N) {
        var str = readLine()!!.split(" ")
        members.add(Member(str[0].toInt(), str[1]))
    }

    members.sortBy { it.age }

    for(i in 0 until N) {
        println("${members[i].age} ${members[i].name}")
    }
}

Kotlin 문서에 의하면 어떤 정렬 알고리즘을 사용하는지는 나와있지 않지만
Kotlin의 sortBy() 함수는 안정 정렬이라고 한다.
따라서 나이 기준으로만 정렬 해 주면 정렬이 완료된다.

코드 보기(Python)
def main():
    n = int(input())
    members = list()

    for i in range(n):
        member = input().split()
        age = int(member[0])
        name = member[1]
        members.append((age, name))

    members.sort(key = lambda age : age[0])

    for i in members:
        print(i[0], i[1])
        
if __name__ == "__main__":
    main()

Python 문서에 의하면 어떤 정렬 알고리즘을 사용하는지는 나와있지 않지만
Python의 sort() 함수는 안정 정렬이라고 한다.
따라서 나이 기준으로 정렬 해 주면 정렬이 완료된다.

코드 보기(Swift)
import Foundation

struct Member {
    var age: Int
    var name: String
}

func main() {
    let n = Int(readLine()!)!
    
    var members = Array<Member>()
    
    for i in 0..<n {
        var member = (readLine()?.split(separator: " "))!
        
        if let age: Int = Int(member[0]) {
            members.append(Member(age: age, name: String(member[1])))
        }
    }
    
    members.sort { first, second in
        return first.age < second.age
    }
    
    for i in 0..<n {
        print(members[i].age, members[i].name)
    }
}

main()

Swift의 sort() 함수는 팀 정렬(Tim Sort)을 이용하므로 안정 정렬이다. (Swift 5 부터 안정 정렬이라고 한다.)
따라서 나이 기준으로만 정렬 해 주면 정렬이 완료된다는 것을 알 수 있다.


물론 안정 정렬을 사용하지 않는 방법도 있다.
입력된 순서대로 가중치를 주어 정렬 기준이 같으면 가중치를 기준으로 정렬 하면 된다.


Tags

#algorithm
Previous Article
BOJ 17386 선분 교차 1
kuper0201

kuper0201

안녕하세요!

Related Posts

BOJ 1009 분산처리
BOJ 1009 분산처리
2023-05-25
1 min

Quick Links

HomeAbout Me

Social Media