[Softeer] 성적 평가 (Python)

 

https://softeer.ai/practice/info.do?idx=1&eid=1309

Softeer

문제

현주는 N명의 인원이 참여하는 프로그래밍 스터디 그룹을 이끌고 있다.

​

현주는 스터디를 위해 대회를 세 개 개최하였고, 모든 구성원이 각 대회에 참여하였다. 참가자는 각 대회에서 0 이상 1,000 이하의 정수인 점수를 얻는다. 한 대회에서 둘 이상의 참가자가 동점이 나오는 경우도 있을 수 있다.

​

현주는 각 대회별 등수 및 최종 등수를 매기고 싶다. 등수는 가장 점수가 높은 사람부터 1등, 2등, ···, N등의 순서대로 붙는다. 만일 동점이 있을 경우 가능한 높은 (등수의 수가 작은) 등수를 부여한다. 즉, 점수가 내림차순으로 10,7,6,6,4의 순서일 경우, 6점을 받은 두 사람은 공동 3등이 되고, 그 다음 순서인 4점을 받은 사람은 5등이 된다. 이 규칙을 다르게 표현하면 다음과 같다: 각 사람마다 “나보다 점수가 큰 사람”의 수를 세어 1을 더한 것이 자신의 등수가 된다. 대회별 등수는 각 대회에서 얻은 점수를 기준으로 하며 최종 등수는 세 대회의 점수의 합을 기준으로 한다.

​

각 참가자의 대회별 등수 및 최종 등수를 출력하는 프로그램을 작성하시오.

​

제약조건

3 ≤ N ≤ 100,000

​

입력형식

첫째 줄에 참가자의 수를 나타내는 정수 N이 주어진다.

이어 세 개의 줄에 각 대회의 결과를 나타내는 N개의 정수가 주어진다. 이중 i번째 정수는 그 대회에서 i번째 사람이 얻은 점수를 의미한다.

​

출력형식

첫 세 개의 줄에는 각 참가자의 대회별 등수를 출력한다. 즉 이중 c번째 줄의 i번째 정수는 c번째 대회에서의 i번째 사람의 등수를 의미한다.

이어 새로운 줄에 같은 형식으로 각 참가자의 최종 등수를 출력한다.

​

입력예제1

3

40 80 70

50 10 20

100 70 30

​

출력예제1

3 1 2

1 3 2

1 2 3

1 2 3

​

입력예제2

3

1000 1000 700

300 100 400

200 400 400

​

출력예제2

1 1 3

2 3 1

3 1 1

1 1 1

​

풀이

n = int(input())
SCORE = [list(map(int, input().split())) for _ in range(3)]
SUM = [0 for _ in range(n)]

def quick_sort(array, start, end):
    if start >= end:
        return

    pivot = start
    left = start + 1
    right = end

    while left <= right:
        while left <= end and array[left] >= array[pivot]:
            left += 1
        while right > start and array[right] <= array[pivot]:
            right -= 1
        if left > right:
            array[pivot], array[right] = array[right], array[pivot]
        else:
            array[left], array[right] = array[right], array[left]

    quick_sort(array, start, right-1)
    quick_sort(array, right+1, end)

for score in SCORE:
    dic = {}
    idx = 0
    origin_score = score
    sorted_score = score.copy()
    quick_sort(sorted_score, 0, n-1)

    for s in sorted_score:
        idx += 1
        if s not in dic:
            dic[s] = idx

    arr = []
    for i in range(len(score)):
        arr.append(dic[origin_score[i]])
        SUM[i] += score[i]
    print(*arr)

dic = {}
idx = 0
origin_sum = SUM
sorted_sum = SUM.copy()
quick_sort(sorted_sum, 0, n-1)
for s in sorted_sum:
    idx += 1
    if s not in dic:
        dic[s] = idx

arr = []
for i in range(len(sorted_sum)):
    arr.append(dic[SUM[i]])
print(*arr)

그냥 단순하게 반복문을 써서 하면 시간초과로 오답이 된다. 따라서 별도의 정렬 알고리즘을 사용해서 풀어야 한다. merge sort를 사용해서 푼 분이 계셔서 그 분의 풀이를 많이 참고해서 풀었다. 그래서 본인은 quick sort를 사용해서 풀어보았다.

​

​

표준 라이브러리 정렬을 사용해도 정답처리로 동작한다. 표준 라이브러리 정렬 또한 복잡도가 O(NlogN)이기 때문인듯?

 

n = int(input())
array = [list(map(int, input().split())) for _ in range(3)]
SUM = [0 for _ in range(n)]

for grade in array:
    sorted_grade = sorted(grade, reverse=True)
    idx = 0
    dic = {}

    for s in sorted_grade:
        idx += 1
        if s not in dic:
            dic[s] = idx

    arr = []
    for i in range(n):
        arr.append(dic[grade[i]])
        SUM[i] += grade[i]
    print(*arr)

sorted_SUM = sorted(SUM, reverse=True)
idx = 0
dic = {}
for s in sorted_SUM:
    idx += 1
    if s not in dic:
        dic[s] = idx

arr = []
for i in range(n):
    arr.append(dic[SUM[i]])

print(*arr)