본문 바로가기

Java/Java 알고리즘 인프런

[Ch.06 - SortSearch] 03. 삽입정렬

반응형
3. 삽입 정렬
 

설명

N개이 숫자가 입력되면 오름차순으로 정렬하여 출력하는 프로그램을 작성하세요.

정렬하는 방법은 삽입정렬입니다.

입력

첫 번째 줄에 자연수 N(1<=N<=100)이 주어집니다.

두 번째 줄에 N개의 자연수가 공백을 사이에 두고 입력됩니다. 각 자연수는 정수형 범위 안에 있습니다.

 

출력

오름차순으로 정렬된 수열을 출력합니다.

예시 입력 1 

6
11 7 5 6 10 9

예시 출력 1

5 6 7 9 10 11

import java.util.Scanner;

public class Main {
	public int[] solution(int n, int[] arr) {
		for (int i = 1; i < n; i++) {
			//for (int j = i - 1; j >= 0; j--) {
			for (int j = 0; j <= i - 1; j++) {
				if (arr[i] < arr[j]) {
					int tmp = arr[i];
					arr[i] = arr[j];
					arr[j] = tmp;
				}
			}
		}
		return arr;
	}

	public static void main(String[] args) {
		Main T = new Main();
		Scanner kb = new Scanner(System.in);
		int n = kb.nextInt();
		int[] arr = new int[n];
		for (int i = 0; i < n; i++) {
			arr[i] = kb.nextInt();
		}
		for (int x : T.solution(n, arr)) {
			System.out.print(x + " ");
		}
	}
}

 

+) 세련된 풀이

import java.util.Scanner;

public class Main {
	// # 삽입정렬
	// 적당한 자리에 A[i]번째를 삽입

	public static void main(String[] args) {
		Main T = new Main();
		Scanner kb = new Scanner(System.in);
		int n = kb.nextInt();
		int[] arr = new int[n];
		for (int i = 0; i < n; i++) {
			arr[i] = kb.nextInt();
		}
		for (int x : T.solution(n, arr))
			System.out.print(x + " ");
		kb.close();
	}

	public int[] solution(int n, int[] arr) {
		for (int i = 1; i <n; i++) {
			//1부터 n까지 반복
			int tmp=arr[i],j;
			
			
			for(j=i-1;j>=0;j--) {
				if(arr[j]>tmp)
					arr[j+1]=arr[j];
				else 
					break;
			}
			arr[j+1]=tmp;
			
		}
		return arr;
	}

}
반응형