코딩테스트

[파이썬] 75. Sort Colors - 리트코드

Gogozzi 2022. 5. 27. 17:47
반응형

문제

빨간색, 흰색 또는 파란색으로 색상이 지정된 개체 가 있는 배열 nums이 주어지면 동일한 색상의 개체가 빨간색, 흰색 및 파란색 순서로 색상이 인접하도록 정렬하는 프로그램을 작성하시오

정수 0, 1, 2 를 사용하여 빨강, 흰색, 파랑을 각각 나타냅니다.

입력

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

출력

Output: [0,0,1,1,2,2]

작성한 코드1

def sortColors(nums: List[int]) -> None:
    nums.sort()

작성한 코드2 - 버블정렬이용

def sortColors(self, nums: List[int]) -> None:
    for i in range(len(nums)):
        for j in range(len(nums)-1-i):
            if nums[j] > nums[j+1]:
                nums[j], nums[j+1] = nums[j+1], nums[j]

https://leetcode.com/problems/sort-colors/

 

Sort Colors - LeetCode

Level up your coding skills and quickly land a job. This is the best place to expand your knowledge and get prepared for your next interview.

leetcode.com

반응형