(Array / String) Easy, Merge Sorted Array

송재호·1일 전
0

https://leetcode.com/problems/merge-sorted-array/description/?envType=study-plan-v2&envId=top-interview-150

1안 - O(nlogn)

원래 문제 의도가 아닌듯?

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        for (int i=0; i<n; i++) {
            nums1[i + m] = nums2[i];
        }
        Arrays.sort(nums1);
    }
}

2안 - O(n)

정확하게는 O(m+n)
둘다 소팅되어있다고 가정하므로 뒤에서부터 투포인터로.

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int i = m - 1;
        int j = n - 1;
        int k = m + n - 1;
        
        while (i >= 0 && j >= 0) {
            if (nums1[i] > nums2[j]) {
                nums1[k--] = nums1[i--];
            } else {
                nums1[k--] = nums2[j--];
            }
        }

        while (j >= 0) {
            nums1[k--] = nums2[j--];
        }
    }
}
profile
식지 않는 감자

0개의 댓글