[백준 알고리즘][파이썬/Python] 7568번: 덩치

문제

시간 제한 : 1 초, 메모리 제한 :  128 MB

 

우리는 사람의 덩치를 키와 몸무게, 이 두 개의 값으로 표현하여 그 등수를 매겨보려고 한다. 어떤 사람의 몸무게가 x kg이고 키가 y cm라면 이 사람의 덩치는 (x,y)로 표시된다. 두 사람 A 와 B의 덩치가 각각 (x,y), (p,q)라고 할 때 x>p 그리고 y>q 이라면 우리는 A의 덩치가 B의 덩치보다 "더 크다"고 말한다. 예를 들어 어떤 A, B 두 사람의 덩치가 각각 (56,177), (45,165) 라고 한다면 A의 덩치가 B보다 큰 셈이 된다. 그런데 서로 다른 덩치끼리 크기를 정할 수 없는 경우도 있다. 예를 들어 두 사람 C와 D의 덩치가 각각 (45, 181), (55,173)이라면 몸무게는 D가 C보다 더 무겁고, 키는 C가 더 크므로, "덩치"로만 볼 때 C와 D는 누구도 상대방보다 더 크다고 말할 수 없다.

 

N명의 집단에서 각 사람의 덩치 등수는 자신보다 더 "큰 덩치"의 사람의 수로 정해진다. 만일 자신보다 더 큰 덩치의 사람이 k명이라면 그 사람의 덩치 등수는 k+1이 된다. 이렇게 등수를 결정하면 같은 덩치 등수를 가진 사람은 여러 명도 가능하다. 아래는 5명으로 이루어진 집단에서 각 사람의 덩치와 그 등수가 표시된 표이다.

 

이름	<몸무게, 키>	덩치 등수
A	<55, 185>	2
B	<58, 183>	2
C	<88, 186>	1
D	<60, 175>	2
E	<46, 155>	5

위 표에서 C보다 더 큰 덩치의 사람이 없으므로 C는 1등이 된다. 그리고 A, B, D 각각의 덩치보다 큰 사람은 C뿐이므로 이들은 모두 2등이 된다. 그리고 E보다 큰 덩치는 A, B, C, D 이렇게 4명이므로 E의 덩치는 5등이 된다. 위 경우에 3등과 4등은 존재하지 않는다. 여러분은 학생 N명의 몸무게와 키가 담긴 입력을 읽어서 각 사람의 덩치 등수를 계산하여 출력해야 한다.

 

입력

첫 줄에는 전체 사람의 수 N이 주어진다. 그리고 이어지는 N개의 줄에는 각 사람의 몸무게와 키를 나타내는 양의 정수 x와 y가 하나의 공백을 두고 각각 나타난다. 단, 2 ≤ N ≤ 50, 10 ≤ x,y ≤ 200 이다.

 

출력

여러분은 입력에 나열된 사람의 덩치 등수를 구해서 그 순서대로 첫 줄에 출력해야 한다. 단, 각 덩치 등수는 공백문자로 분리되어야 한다.

 

입출력 예

 

제출 코드

n = int(input())
lst = [list(map(int, input().split())) for i in range(n)]
ans = [1 for i in range(n)]

for i in range(n):
    cnt = 0
    for j in range(n):
        if i == j:
            continue
        if lst[i][0] < lst[j][0] and lst[i][1] < lst[j][1]:
            cnt += 1
    ans[i] += cnt

for i in ans:
    print(i, end = ' ')

 

문제의 핵심

# 1

메모리가 128MB나 주어지고 사람 수가 2명에서 50명까지 밖에 안 된다면 브루트 포스로 구현하여 pypy3로 제출하면 될 것이라는 생각을 해야 한다. 그러면 딱히 먼저 몸무게 순으로 정렬한 후 키를 비교한다든가 하는 짓을 귀찮게 할 필요가 없다.

 

라인 별 풀이

(1) line 2

입력받는 각 사람의 몸무게와 키를 리스트로 저장하여 2차원 리스트로 만든다.

 

(2) line 3

답으로 출력할 answer 리스트를 선언하는데, 인원수만큼 1로 초기화한다. 나중에 자기보다 '덩치가 큰 사람'의 수만큼 각 element에 더해줄 것인데, 만약 자기보다 덩치가 큰 사람이 없는 경우는 덩치의 등수가 1등이므로 1로 초기화하는 것이다.

 

(3) line 5

모든 사람을 각각 순회하며 해당 사람과 다른 사람들의 덩치를 비교해야 한다. 그래서 n번 반복한다.

 

(4) line 6

(2)에서 말했듯 자기보다 덩치가 큰 사람의 수만큼 카운트해줘야 하므로 변수 cnt를 선언하고 0으로 초기화한다.

 

(5) line 7 ~ 9

다른 사람들과 덩치를 비교해야 하므로 엄밀히 말하면 n-1로 반복을 돌려야 하는 것이 맞다. 하지만 자신보다 index가 앞에 있는 사람들과 자신보다 index가 뒤에 있는 사람들과 비교해야 하므로, 위 코드처럼 n번 반복한 후 비교할 대상이 자신이면 continue로 흘려보내는 것이 훨씬 간단하다고 생각된다.

 

굳이 복잡하게 짜자면 for i in range(0, i) 과 for (i+1, n) 반복문을 두 번 돌리면 되겠지만 권장하지 않는다.

 

(6) line 10, 11

문제에서는 '두 사람 A 와 B의 덩치가 각각 (x,y), (p,q)라고 할 때 x>p 그리고 y>q 이라면 우리는 A의 덩치가 B의 덩치보다 "더 크다"고 말한다'라고 힌트를 줬다. 그냥 그대로 코딩을 하면 된다. 비교하는 대상의 덩치가 더 클 경우, cnt에 1을 더해준다.

 

(7) line 12

다른 사람들과 비교가 끝났다면 해당 사람의 등수를 answer 리스트에 반영해준다.

 

(8) line 14, 15

문제에서 요구하는 출력 형태로 답을 출력한다.

 

 

 

Source : https://www.acmicpc.net/problem/7568