[Python-CodingTest] 2-2. K번째 수

K 번째 수

문제 정리

입력

2 // 테스트 케이스 수 (T)
6 2 5 3 // N s e k
5 2 7 3 8 9 // N개의 숫자로 이루어진 수열
15 3 10 3 // N s e k
4 15 8 16 6 6 17 3 10 11 18 7 14 7 15 // N개의 숫자로 이루어진 수열

처리 과정

1. 2~5번째: 2 7 3 8
2. 오름차순: 2 3 7 8
3. 3번째 수: 7

출력

#1 7
#2 6

풀이

import sys
sys.stdin = open("./input/in2.txt", "rt")

T = int(input())
for t in range(T):
    n,s,e,k = map(int, input().split())
    lst = list(map(int, input().split()))

    lst = lst[s-1:e] # s-1번 익덱스부터 e-1번 인덱스까지 추출
    lst.sort() # 오름차순 정렬
    print("#%d %d" %(t+1, lst[k-1])) 

정리

• 테스트 케이스의 개수가 주어질 때는 for문 이용
• list[1:5]라면 1번 인덱스부터 4번 인덱스까지 추출
• print() 출력 형식 지정: print("%d %d" %(a, b))

---

💛 개인 공부 기록용 블로그입니다. 👻

맨 위로 이동하기