[구름LEVEL] 이진수 정렬

[구름LEVEL] 이진수 정렬

https://level.goorm.io/exam/195687/%EC%9D%B4%EC%A7%84%EC%88%98-%EC%A0%95%EB%A0%AC/quiz/1

문제 설명

· N개의 10진수 정수가 주어질 때, 아래 기준에 따라 정렬

· 10진수 정수를 2진수로 나타냈을 때, 2진수의 1의 개수를 기준으로 내림차순 정렬

· 1의 개수가 같을 경우, 원래 10진수를 기준으로 내림차순 정렬

· 앞에서 K번째 위치한 수를 반환

2진수로 나타냈을 때의 1의 개수, 같다면 원래 수의 크기를 기준으로 수의 배열을 내림차순 정렬하는 문제

접근 방법

· bin()을 통해 2진수로 바꿔준 후, str객체의 count로 1의 개수를 세기

· 해당 함수를 sorted의 첫번째 key로 넘겨주기

· 1의 개수가 같을 때, 원래 크기로 정렬하기 때문에 원래 수를 두번째 key로 넘겨주기

· reverse를 통해 내림차순 정렬한 후, 인덱스는 0부터 시작이므로 K-1 인덱스의 값 반환

알고리즘 설계 및 구현

N, K = input().split()
arr = map(int, input().split())
answer = sorted(arr, key = lambda x: (str(bin(x)).count('1'), x), reverse=True)
print (answer[int(K)-1])

결과