Algorithm - K번째 수

 

알고리즘 입문 수업을 듣고 중요한 내용을 정리했습니다.
개인 공부 후 자료를 남기기 위한 목적이므로 내용 상에 오류가 있을 수 있습니다.

 

문제

[K번째 수]

 

N개의 숫자로 이루어진 숫자열이 주어지면 해당 숫자열중에서 s번째부터 e번째까지의 수를 오름차순으로 정렬했을 때, K번째로 나타나는 숫자를 출력하는 프로그램을 작성하시오.

 

 

*입력 설명

첫 번째 줄에 테스트 케이스 T(1 ≤ T ≤ 10)가 주어진다.

각 케이스 마다 첫 번째 줄에는 자연수 N(5 ≤ N ≤ 500)과 s, e, k가 차례로 주어진다.

각 케이스 마다 두 번째 줄에는 N개의 숫자가 차례로 주어진다.

 

*출력 설명

각 케이스 마다 K번째 수를 출력한다.

 

 

---

 

 

풀이(Python)

답안

import sys
sys.stdin = open('AA/input_2.txt', 'rt')

T = int(input())
for t in range(T):
    n, s, e, k = map(int, input().split())
    a = list(map(int, input().split()))
    a = a[s-1:e]
    a.sort()
    print('#%d %d' % (t + 1, a[k-1]))
    
    
# 출력 : 
# #1 7
# #2 6

 

input_2.txt(입력)

2
6 2 5 3
5 2 7 3 8 9
15 3 10 3
4 15 8 16 6 6 17 3 10 11 18 7 14 7 15

 

 

중요내용

1. input() 함수를 통해 input_2.txt 파일에서 코드를 입력받는 경우, input() 함수는 해당 파일의 코드를 한 줄씩만 받아온다.
2. 따라서 input_2.txt 파일의 5줄 코드를 모두 받아오기 위해서는, 5번의 input() 함수가 실행되어야 한다.
3. 이러한 경우, 반복문을 활용하면  input_2.txt 파일의 코드를 편리하게 (한 줄씩)받아올 수 있다.