728x90

23881번: 알고리즘 수업 - 선택 정렬 1
첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 교환 횟수 K(1 ≤ K ≤ N)가 주어진다. 다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 109)
www.acmicpc.net
문제
오늘도 서준이는 선택 정렬 수업 조교를 하고 있다. 아빠가 수업한 내용을 학생들이 잘 이해했는지 문제를 통해서 확인해보자.
N개의 서로 다른 양의 정수가 저장된 배열 A가 있다. 선택 정렬로 배열 A를 오름차순 정렬할 경우 K 번째 교환되는 수를 구해서 우리 서준이를 도와주자.
크기가 N인 배열에 대한 선택 정렬 의사 코드는 다음과 같다.
selection_sort(A[1..N]) { # A[1..N]을 오름차순 정렬한다. for last <- N downto 2 { A[1..last]중 가장 큰 수 A[i]를 찾는다 if (last != i) then A[last] <-> A[i] # last와 i가 서로 다르면 A[last]와 A[i]를 교환 } }
입력
첫째 줄에 배열 A의 크기 N(5 ≤ N ≤ 10,000), 교환 횟수 K(1 ≤ K ≤ N)가 주어진다.
다음 줄에 서로 다른 배열 A의 원소 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 109)
출력
K 번째 교환되는 두 개의 수를 작은 수부터 한 줄에 출력한다. 교환 횟수가 K 보다 작으면 -1을 출력한다.
풀이
의사코드를 이용하여 선택정렬을 구현한다.
교환이 생길때마다 cnt변수를 이용하여 카운팅을 하고 cnt가 K와 동일해지는 순간 교환되는 값을 출력한다.
코드
import java.io.*; import java.util.*; public class Main { public static void main(String[] args) throws Exception{ BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(br.readLine()); int n= Integer.parseInt(st.nextToken()), k=Integer.parseInt(st.nextToken()), cnt=0; int [] a = new int [n]; st = new StringTokenizer(br.readLine()); for(int i=0;i<n;i++) a[i]=Integer.parseInt(st.nextToken()); for(int i=n-1;i>0;i--) { int m=0, idx=0; for(int j=0;j<=i;j++) { if(m<a[j]) { m=a[j]; idx=j; } } if(a[i]!=m) { int t=a[i]; a[i]=a[idx]; a[idx]=t; cnt++; if(cnt==k) { System.out.println(Math.min(a[i], a[idx])+" "+Math.max(a[i], a[idx])); break; } } } if(cnt<k) System.out.println("-1"); } }
728x90
728x90
'Algorithm > Baekjoon' 카테고리의 다른 글
Baekjoon 3060 욕심쟁이 돼지 JAVA (1) | 2022.10.04 |
---|---|
Baekjoon 14728 벼락치기 JAVA (1) | 2022.10.04 |
Baekjoon 5893 17배 JAVA (0) | 2022.09.20 |
Baekjoon 5347 LCM JAVA (0) | 2022.09.18 |
Baekjoon 21774 가희와 로그 파일 JAVA (0) | 2022.09.18 |
댓글