Gidhub BE Developer

LeetCode : 88. Merge Sorted Array

2020-12-12
goodGid

88. Merge Sorted Array

Problem

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Example

Input:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6],       n = 3

Output: [1,2,2,3,5,6]

[1] Code (20. 12. 12)

class Solution {
public void merge(int[] nums1, int m, int[] nums2, int n) {

        int[] ansArray = new int[m + n];

        int i = 0;
        int j = 0;

        for (int idx = 0; idx < m + n; idx++) {
            if (i == m) {
                ansArray[idx] = nums2[j];
                j++;
                continue;
            } else if (j == n) {
                ansArray[idx] = nums1[i];
                i++;
                continue;
            }

            if (nums1[i] == nums2[j]) {
                ansArray[idx] = nums1[i];
                ansArray[idx + 1] = nums1[i];
                idx++;
                i++;
                j++;
            } else if (nums1[i] < nums2[j]) {
                ansArray[idx] = nums1[i];
                i++;
            } else {
                ansArray[idx] = nums2[j];
                j++;
            }
        }

        for (int k = 0; k < m + n; k++) {
            nums1[k] = ansArray[k];
        }
    }
}
  • 아주 직관적인 코드

[2] Code (20. 12. 12)

class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {

        int i = m - 1;
        int j = n - 1;

        for (int k = m + n - 1; k >= 0; k--) {
            if (i < 0) {
                nums1[k] = nums2[j--];
            } else if (j < 0) {
                nums1[k] = nums1[i--];
            } else if (nums1[i] < nums2[j]) {
                nums1[k] = nums2[j--];
            } else {
                nums1[k] = nums1[i--];
            }
        }
    }
}
  • m과 n의 값을 사용하여

    뒤에서 부터 값을 채워 넣는다.

    그리고 이게 가능한 이유는 정렬된 상태로 array가 input으로 들어오기 때문이다.

    two sorted integer arrays


[3] Code (22. 03. 13) (x)

// Runtime: 0 ms
// Memory Usage: 43.4 MB
// Ref : https://leetcode.com/submissions/detail/658848054
class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {

        int idx = m + n - 1;

        while (true) {
            if (m == 0 && n == 0) {
                break;
            } else if (m == 0) {
                nums1[idx] = nums2[n - 1];
                idx--;
                n--;
                continue;
            } else if (n == 0) {
                nums1[idx] = nums1[m - 1];
                idx--;
                m--;
                continue;
            }

            if (nums1[m - 1] >= nums2[n - 1]) {
                nums1[idx] = nums1[m - 1];
                idx--;
                m--;
            } else {
                nums1[idx] = nums2[n - 1];
                idx--;
                n--;
            }
        }
    }
}
  • Reference Code를 보고 내 코드를 보면 한없이 작아지는 느낌… ㅠ_ㅠ

Reference Code

Code 1

// Runtime: 1 ms
// Memory Usage: 39 MB
// Ref : https://leetcode.com/submissions/detail/429851648
class Solution {
    public void merge(int[] nums1, int m, int[] nums2, int n) {

        int i = m - 1;
        int j = n - 1;

        for (int k = m + n - 1; k >= 0; k--) {
            if (i < 0) {
                nums1[k] = nums2[j--];
            } else if (j < 0) {
                nums1[k] = nums1[i--];
            } else if (nums1[i] < nums2[j]) {
                nums1[k] = nums2[j--];
            } else {
                nums1[k] = nums1[i--];
            }
        }
    }
}
  • 내가 푼 코드에서 idx 변수 역할을 int k가 대신한다.

    위 코드가 내 코드보다 깔끔하다.


Review

  • 조금 더 깔끔한 코딩을 하고 싶다.

Reference


Recommend

Index