[Algorithm] 프로그래머스 K 번째 수

 

 

# 내가 푼 알고리즘 
def solution(array, commands):
b=[]
for i in range(0,len(commands)):	
	array.sort()
	answer=array[ (commands[i][0])-1: (commands[i][1])]
	
	#print(answer)
	b.append(answer[ (commands[i][2])-1 ])
	print(b)

	#print(answer)
return answer
]

데이터를 받아서 sort하고 거기서 command에서 나온 값을 넣어서 찾는 식으로 하였다.

#남이 푼 알고리즘
def solution2(array, commands):
	return list(map(lambda x:sorted(array[x[0]-1:x[1]])[x[2]-1], commands))

array = [1, 5, 2, 6, 3, 7, 4]
commands = [[2, 5, 3], [4, 4, 1], [1, 7, 3]]


solution(array, commands)

이런식으로 한줄에 끝낼수도있다.