[Algorithm][Python] K번째 작은수

[문제]  K번째 작은수

어떤 자연수 p와 q가 있을 때, 만일 p를 q로 나누었을 때 나머지가 0이면 q는 p의 약수이다.

N개의 숫자로 이루어진 숫자열이 주어지면 해당 숫자열중에서 s번째부터 e번째 까지의 수 중 
k번째로 작은 수를 출력하는 프로그램을 작성하세요.

 

입력 설명

첫째 줄에 N과 K가 빈칸을 사이에 두고 주어진다. N은 1 이상 10,000 이하이다. K는 1 이상 N 
이하이다.첫 번째 줄에 테스트 케이스 T(1<=T<=10)이 주어집니다.
각 케이스별 
첫 번째 줄은 자연수 N(5<=N<=500), s, e, k가 차례로 주어진다.
두 번째 줄에 N개의 숫자가 차례로 주어진다.

 

출력 설명

첫째 줄에 N의 약수들 중 K번째로 작은 수를 출력한다. 만일 N의 약수의 개수가 K개보다 적어서 
K번째 약수가 존재하지 않을 경우에는 -1을 출력하시오.

 

입력예제 1

6 3 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

#1 7
#2 6

 

문제 접근

1. 테스트 케이스의 갯수인 T를 입력받고 T의 갯수만큼 for문을 돌린다.
2. N의 갯수인 정수들을 리스틀 받아서 s-1부터 e까지 슬라이싱 해주고 오름차순 정렬을 해준다.
3. 테스트 케이스의 수와 result에서 k에 해당하는 값은 k-1로 접근해서 출력한다.

 

풀이

 

나의 코드

• 현재 예시는 s(2번째)부터 e(5번째)를 가져와야 한다.
  • 파이썬에서 인덱스의 시작 번호는 0부터 시작이다. 그래서 인덱스로는 1~4번의 값들을 가져와야 한다.
  • result[s-1 : e] => result[2-1 : 5] => result[1: 5]
    • result리스트에서 1에서 5까지 즉 1~4의 인덱스의 값을 가져온다. 

# T : 테스트 케이스 수 입력
T = int(input())

# 테스트 케이스만큼 for문 돌기
for t in range(T):
    N, s, e, k = map(int, input().split())
    
    # N의 갯수만큼 입력을 list로 받고  
    # list를 s-1부터 e까지 슬라이싱 후 
    # sorted()로 오름차순 정렬 후 result 변수에 넣어줍니다.
    result = sorted(list(map(int, input().split()))[s-1:e])

    print(f'#{t+1} {result[k-1]}')

 

정답 코드

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]))