[BOJ/Python] 11582_치킨 TOP N || Divide & Conquer

11582번: 치킨 TOP N

백준 11582 파이썬
문제 설명

인하대 주변 치킨칩의 맛의 정도를 측정해 수치화하는 동아리 C.T.P(Chicken Tastes Perfect)의 회장 민호는 치킨집의 맛의 수치를 감소하지 않는 순으로 정렬을 하고 싶었다. 하지만 치킨집이 너무 많아 혼자 정렬을 하기에는 많은 시간이 걸려 C.T.P 회원들을 활용하기로 했다. 치킨집이 N개 있다고 가정을 하자. N개의 치킨의 수치를 무작위로 놓은 뒤 N/2명의 C.T.P 회원이 차례대로 2개의 치킨집을 선택해  정렬을 한다. 그 뒤 N/4명이 차례대로 바로 전 단계의 사람이 정렬한 두 개의 그룹을 차례대로 선택 하여 치킨집을 정렬을 한다. 계속해서 N/8명, N/16명이 정렬을 진행하다가 마지막 사람이 두 개의 정렬된 그룹을 합병하여 작업을 완료한다.

예를 들어 8개의 치킨집의 점수가 (1, 5, 2, 4, 2, 9, 7, 3)과 같을 때, 첫 번째로 4명의 회원이(1, 5), (2, 4), (2, 9), (7, 3)을 각각 정렬하게 되면 (1, 5), (2, 4), (2, 9), (3, 7)이 되고, 두 번째로 2명의 회원이 ((1, 5), (2, 4))와 ((2, 9), (3, 7))을 정렬하면 (1, 2, 4, 5), (2, 3, 7, 9)가 되고, 최종적으로 1명의 회원이 ((1, 2, 4, 5), (2, 3, 7, 9))을 정렬하여 (1, 2, 2, 3, 4, 5, 7, 9)을 만들게 된다.

작업을 진행하던 중 문득 민호는 작업의 중간단계가 궁금해졌다. 현재 단계에서 k명의 회원이 정렬을 진행할 때 정렬을 마친 뒤 결과를 출력해라.

입력 )

첫 번째 줄에 치킨집의 개수 N(4 ≤ N ≤ $ 2^{20}) $)이 주어진다. N은 항상 2의 거듭제곱 꼴이다.

두 번째 줄에는 N개의 치킨집의 맛의 수치들이 빈칸을 구분으로 주어지며 이 값은 10억보다 작거나 같은 자연수 또는 0이다.

세 번째 줄에는 현재 정렬을 진행중인 회원들의 수 k(1 ≤ k < N)가 주어진다. k 또한 2의 거듭제곱 꼴이다.

출력 ) 
현재 단계에서 k명이 정렬을 진행한다고 할 때, 현재 단계가 완료한 상태를 출력하라.

예시 )

 

📌 문제 풀이

시간 초과된 문제 풀이
k번 for문을 돌리며, 분할된 수들을 합쳐서 정렬을 진행해주었다.

# 불필요한 k번의 연산으로 시간 초과

n = int(input())
arr = list(map(int, input().split()))
k = int(input())


for i in range(k):
    step = 2 * (i + 1)
    for i in range(0, len(arr), step):
        arr[i:i+step] = sorted(arr[i:i+step])

for i in arr:
    print(i, end=" ")

생각해보니, sorted 함수를 이용하여 계산을 진행하기 때문에, k번 합쳐가며 정렬할 필요가 없었다...
따라서, 원하는 범위만 정렬이 진행될 수 있도록 수정해주었다.
 

✅ [ CODE ]

n = int(input())
arr = list(map(int, input().split()))
k = int(input())

step = n // k
for i in range(0, len(arr), step):
    arr[i:i+step] = sorted(arr[i:i+step])

for i in arr:
    print(i, end=" ")