728x90
문제 설명
문제
오늘도 서준이는 버블 정렬 수업 조교를 하고 있다. 아빠가 수업한 내용을 학생들이 잘 이해했는지 문제를 통해서 확인해보자.
N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 버블 정렬로 배열 A를 오름차순 정렬할 경우 K 번째 교환되는 수를 구해서 우리 서준이를 도와주자.
크기가 N인 배열에 대한 버블 정렬 의사 코드는 다음과 같다.
bubble_sort(A[1..N]) { # A[1..N]을 오름차순 정렬한다.
for last <- N downto 2
for i <- 1 to last - 1
if (A[i] > A[i + 1]) then A[i] <-> A[i + 1] # 원소 교환
}
입력
첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 교환 횟수 K(1 ≤ K ≤ N2)가 주어진다.
다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 109)
출력
K 번째 교환되는 두 개의 수를 작은 수부터 한 줄에 출력한다. 교환 횟수가 K 보다 작으면 -1을 출력한다.
예제 입력 1
6 10
4 6 5 1 3 2
예제 출력 1
2 4
4 6 5 1 3 2 -> 4 5 6 1 3 2 -> 4 5 1 6 3 2 -> 4 5 1 3 6 2 -> 4 5 1 3 2 6 -> 4 1 5 3 2 6 -> 4 1 3 5 2 6 -> 4 1 3 2 5 6 -> 1 4 3 2 5 6 -> 1 3 4 2 5 6 -> 1 3 2 4 5 6 -> 1 2 3 4 5 6. 총 11회 교환이 발생하고 열 번째 교환은 2와 4이다.
예제 입력 2
6 12
4 6 5 1 3 2
예제 출력 2
-1
교환 횟수 11이 K 보다 작으므로 -1을 출력한다.
풀이
import sys
N, K = map(int, input().split())
A = list(map(int, input().split()))
k = 0
flag = True
for i in range(N-1):
for j in range(N-1-i):
if A[j] > A[j+1]:
A[j], A[j+1] = A[j+1], A[j]
k += 1
if k == K:
print("%d %d" % (A[j], A[j+1]))
flag = False
break
if not flag:
break
if flag:
print(-1)
- Python3으로 하면 시간 초과 뜸. PyPy3으로 해야 시간초과 안됨.
- PyPy3은 자주 쓰이는 코드를 캐싱하는 기능이 있다. 따라서 복잡한 코드(반복이 많은 경우)에서 Python3에 비해 실행속도가 빠르다.
반응형
'알고리즘 > 백준' 카테고리의 다른 글
[백준 / 이진 탐색] 10815 : 숫자 카드 (python) (0) | 2024.04.01 |
---|---|
[백준 / DFS, BFS] 2606 : 바이러스 (python) (0) | 2024.04.01 |
[백준 / 정렬] 23881 : 알고리즘 수업 - 선택 정렬 1 (python) (0) | 2024.03.24 |
[백준 / 정렬] 1377 : 버블 소트 (python) (1) | 2024.03.22 |
[백준 / 큐] 10828 : 스택 (python) (0) | 2024.03.19 |