[Greedy Algorithm] BOJ1946 - 신입사원

문제

언제나 최고만을 지향하는 굴지의 대기업 진영 주식회사가 신규 사원 채용을 실시한다. 인재 선발 시험은 1차 서류심사와 2차 면접시험으로 이루어진다. 최고만을 지향한다는 기업의 이념에 따라 그들은 최고의 인재들만을 사원으로 선발하고 싶어 한다.

그래서 진영 주식회사는, 다른 모든 지원자와 비교했을 때 서류심사 성적과 면접시험 성적 중 적어도 하나가 다른 지원자보다 떨어지지 않는 자만 선발한다는 원칙을 세웠다. 즉, 어떤 지원자 A의 성적이 다른 어떤 지원자 B의 성적에 비해 서류 심사 결과와 면접 성적이 모두 떨어진다면 A는 결코 선발되지 않는다.

이러한 조건을 만족시키면서, 진영 주식회사가 이번 신규 사원 채용에서 선발할 수 있는 신입사원의 최대 인원수를 구하는 프로그램을 작성하시오.

---

입력

2
5
3 2
1 4
4 1
2 3
5 5
7
3 6
7 3
4 2
1 4
5 7
2 5
6 1

출력

4
3

---

시행착오를 많이 겪었다.

첫번째, 문제를 이해하지 못해서이다. 나만 문제가 너무 축약되어있다고 생각한 건가?

두번째, 이중 포문이 많이 들어가서 시간초과가 계속 떴다. 4% 이후로 올라가지가 않았다.

​

먼저 문제를 이해해보자

3 2
1 4
4 1
2 3
5 5

일단 입력 받은 서류점수와 면접점수들을 서류점수 기준으로 정렬한다.

1 4 <---
2 3
3 2
4 1
5 5

 

처음에 머리로 이렇게 이해했다. 첫번째 등수는 서류 점수가 1등이기 때문에 무조건 합격이고, 가장 마지막 등수는 서류점수가 꼴찌이기 때문에 무조건 불합격이라는 것을 염두에 두자.

알고리즘을 생각했을 때, 현재까지 가장 높은 등수인 4를 max에 저장한다.

1 4
2 3 <---
3 2
4 1
5 5

3과 비교했을 때, 4가 더 크기 때문에 3은 합격이다. 3으로 max를 갱신한다. 다음으로 넘어가 2와 비교했을 때, 3이 더 크기 때문에 2는 합격이다. 2를 max로 갱신한다. 이렇게 계속 하면 합격자와 불합격자를 나눌 수 있다.

​

---

소스코드

import sys

n = int(input())
answer = []
count = 1
for i in range(n):
  it = int(input())
  rank = sorted([list(map(int, sys.stdin.readline().strip().split())) for _ in range(it)], key = lambda x: x[0])
  max = rank[0][1]
  for j in range(1, it):
    if(rank[j][1] < max):
      count += 1
      max = rank[j][1]
  answer.append(count)
  count = 1

for i in answer:
  print(i)

시간초과가 난 이유

1. max를 생각해놓고 막상 코드에서 안 썼다.

2. input()으로 입력을 받았더니 시간초과가 나서 sys.stdin.readline().strip()을 애용해야 할 것 같다.

​

---

실패한 코드 1

n = int(input())
answer = []
count = 1
for i in range(n):
  it = int(input())
  rank = [list(map(int, input().split())) for _ in range(it)]
  rank.sort()
  max = rank[0][1]
  for j in range(1, it):
    if(rank[j-1][1] > rank[j][1]):
      count += 1
      max = rank[j-1][1]
  answer.append(count)
  count = 1


for i in answer:
  print(i)

실패한 코드 2

n = int(input())
answer = []
count = 1
for i in range(n):
  it = int(input())
  rank = [list(map(int, input().split())) for _ in range(it)]
  rank.sort()
  max = rank[0][1]
  for j in range(1, it):
    if(rank[j][1] < max):
      count += 1
      max = rank[j][1]
  answer.append(count)
  count = 1

for i in answer:
  print(i)