알고리즘

[LeetCode] 88. Merge Sorted Array

이준언 2024. 9. 13. 16:27

안녕하세요!

 

오늘부터는 LeetCode를 통해 알고리즘 문제를 풀어보려고 합니다.

 

오늘은 88번 Merge Sorted Array 문제를 풀어보겠습니다.

 

문제는 다음과 같습니다.

 

 

 

정수를 포함하고 있는 두 array를 합친 후 정수의 크기를 오름차순 sorting하는 문제였습니다.

 

정수 배열인 nums1, nums2가 있고, 각 배열의 원소의 개수는 m, n으로 표현하고 있습니다.

 

nums1을 수정해서 답이 출력되도록 문제를 풀어야 했습니다. 

 

제가 문제 풀이를 위해 접근한 방식은

 

먼저 nums1에 nums2의 원소를 모두 추가한 후, nums1의 원소를 오름차순 정렬하는 방법입니다.

 

코드는 다음과 같습니다.

 

class Solution(object):
    def merge(self, nums1, m, nums2, n):
        """
        :type nums1: List[int]
        :type m: int
        :type nums2: List[int]
        :type n: int
        :rtype: None
        """
        # nums1에 nums2의 원소를 붙여넣기
        idx = 0 
        for i in range(m, len(nums1)):
            nums1[i] = nums2[idx]  
            idx += 1
        
        # nums1을 오름차순 정렬
        nums1.sort()
        
        # 결과 출력
        print(nums1)

 

제출했더니 다음과 같은 결과가 나왔습니다.

 

모든 의견을 환영합니다. 더 좋은 해결 방법이 있다면 알려주시면 감사하겠습니다!!!!