본문 바로가기

카테고리 없음

leetcode - sort Colors

문제

Given an array nums with n objects colored red, white, or blue, sort them in-place so that objects of the same color are adjacent, with the colors in the order red, white, and blue.
We will use the integers 0, 1, and 2 to represent the color red, white, and blue, respectively.

You must solve this problem without using the library's sort function.

예시
Example 1:
Input: nums = [2,0,2,1,1,0]
Output: [0,0,1,1,2,2]

Example 2:
Input: nums = [2,0,1]
Output: [0,1,2]

 

해설:

빨간, 흰색, 청색으로 이루어진 n개의 요소들이 있는 배열들을 동일한 색상별로 정렬하는 문제이다. 빨강은 0, 흰색은 1, 청색은 2로 표시하면 되고 최종적으로 빨강, 흰색, 청색 순으로 정렬된 배열을 도출하면 된다.

class Solution {
    public void sortColors(int[] nums) {
        int minIndex;
        
        for(int i = 0; i < nums.length; i++){
            //최소값 구하기
            //현재 > 다음 -> 다음 최소값
            // 다음 > 다다음 -> 다다음 최소값
            minIndex = i;
            int temp = 0; 
            
            for(int j = i+1; j<nums.length; j++){
                if(nums[minIndex] > nums[j] ){
                    minIndex = j;
                }
            }
            
            temp = nums[i];
            nums[i] = nums[minIndex];
            nums[minIndex] = temp;
        }
    }
}