본문 바로가기

알고리즘

정렬 알고리즘

class Main {
    public static void main(String[] args) {
        int n = 10;
        int[] arr = {7, 5, 9, 0, 3, 1, 6, 2, 4, 8};

        // 선택 정렬
        for (int i = 0; i < n; i++) {
            int minIdx = i;     // 가장 작은 원소의 인덱스
            for (int j = i + 1; j < n; j++) {
                if (arr[minIdx] > arr[j]) {
                    minIdx = j;
                }
            }
            // 스와프
            int tmp = arr[i];
            arr[i] = arr[minIdx];
            arr[minIdx] = tmp;
        }

        // 삽입 정렬
        for (int i = 1; i < n; i++) {
            for (int j = i; j > 0; j--) {
                if (arr[j] < arr[j - 1]) {
                    int tmp = arr[j];
                    arr[j] = arr[j - 1];
                    arr[j - 1] = tmp;
                }
                // 자기보다 작은 데이터를 만나면 그 위치에서 멈춤
                else break;
            }
        }

        // 퀵 정렬
        quickSort(arr, 0, n - 1);
        
        // 계수 정렬
        int n2 = 15;
        int maxValue = 9;
        // 모든 원소의 값이 0보다 크거나 같다고 가정
        int[] arr2 = {7, 5, 9, 0, 3, 1, 6, 2, 9, 1, 4, 8, 0, 5, 2};
        // 모든 범위를 포함하는 배열 선언 (모든 값은 0으로 초기화)
        int[] cnt = new int[maxValue + 1];

        for (int i = 0; i < n2; i++) {
            cnt[arr2[i]] += 1;      // 각 데이터에 해당하는 인덱스의 값 증가
        }
        for (int i = 0; i <= maxValue; i++) {   // 배열에 기록된 정렬 정보 확인
            for (int j = 0; j < cnt[i]; j++) {
                System.out.println(i + " ");    // 띄어쓰기 기준 등장한 횟수만큼 인덱스 출력
            }
        }

    }

    public static void quickSort(int[] arr, int start, int end) {
        if(start >= end) return;    // 원소가 1개인 경우 종료
        int pivot = start;      // 피벗은 첫 번째 원소
        int left = start + 1;
        int right = end;

        while (left <= right) {
            // 피벗보다 큰 데이터를 찾을 때까지 반복
            while(left <= end && arr[left] <= arr[pivot]) left++;
            // 피벗보다 작은 데이터를 찾을 때까지 반복
            while(right > start && arr[right] >= arr[pivot]) right--;
            // 엇갈렸다면 작은 데이터와 피벗을 교체
            if (left > right) {
                int tmp = arr[pivot];
                arr[pivot] = arr[right];
                arr[right] = tmp;
            }
            // 엇갈리지 않았다면 작은 데이터와 큰 데이터를 교체
            else {
                int tmp = arr[left];
                arr[left] = arr[right];
                arr[right] = tmp;
            }
        }
        // 분할 이후 왼쪽 부분과 오른쪽 부분에서 각각 정렬 수행
        quickSort(arr, start, right - 1);
        quickSort(arr, right + 1, end);
    }
}

'알고리즘' 카테고리의 다른 글

[백준] 인구 이동  (0) 2025.02.28
[백준] 감시 피하기  (0) 2025.02.27
[백준] 연산자 끼워넣기  (0) 2025.02.25
[프로그래머스] 괄호 변환  (0) 2025.02.25
[백준] 경쟁적 전염  (0) 2025.02.24