67차 1번 문제풀이

Author: kingjinyong

live6/test67/문제1/김진용.py

@@ -0,0 +1,50 @@
+def merge_sort(A, p, r):    # A[p~r]을 오름차순 정렬.
+    if p < r:
+        q = (p + r) // 2    # q는 p, r의 중간 지점
+        merge_sort(A, p, q)     # 전반부 정렬
+        merge_sort(A, q + 1, r) # 후반부 정렬
+        merge(A, p, q, r)       # 병합
+
+# A[p~q]와 A[(q+1) ~ r] 을 병합해 A[p~r]을 오름차순 정렬된 상태로 만든다.
+# A[p~q]와 A[(q+1) ~ r]은 이미 오름차순으로 정렬되어 있다.
+def merge(A, p, q, r):
+    global cnt, res
+    i = p
+    j = q + 1
+    tmp = []
+
+    while i <= q and j <= r:
+        if A[i] <= A[j]:
+            tmp.append(A[i])
+            i += 1
+        else:
+            tmp.append(A[j])
+            j += 1
+
+    while i <= q: # 왼쪽 배열 부분이 남은 경우
+        tmp.append(A[i])
+        i += 1
+
+    while j <= r: # 오른쪽 배열 부분이 남은 경우
+        tmp.append(A[j])
+        j += 1
+
+    i = p
+    t = 0
+
+    while i <= r:   # A[p~r]에 저장
+        A[i] = tmp[t]
+        cnt += 1
+        if cnt == K:
+            res = A[i]
+            break;
+        i += 1
+        t += 1
+
+N, K = map(int, input().split())
+A = list(map(int, input().split()))
+
+cnt = 0
+res = -1
+merge_sort(A, 0, N - 1)
+print(res)