88. Combine dos matrices ordenadas y recorra una vez para reducir las variables temporales

tema

Dados dos arreglos de enteros ordenados nums1 y nums2, por favor combine nums2 en nums1 para hacer nums1 un arreglo ordenado.

Descripción:

Inicialice el número de elementos de nums1 y nums2 am y n respectivamente.
Puede suponer que nums1 tiene suficiente espacio (el tamaño del espacio es mayor o igual que m + n) para almacenar los elementos en nums2.

Ejemplo:

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

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

 solución:

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        if (0 == n)
            return;

        int index = m+n-1, mi = m-1, ni = n-1;
        for (; index >= 0 && mi >= 0 && ni >= 0; --index) {
            if (nums1[mi] > nums2[ni]) {
                nums1[index] = nums1[mi];
                --mi;
            } else {
                nums1[index] = nums2[ni];
                --ni;
            }  
        }
        while (ni >= 0)
            nums1[index--] = nums2[ni--];
    }
};

Una implementación más simplificada es la siguiente, sin introducir variables adicionales.

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int len = nums1.size();
        if (len != m+n || n == 0)
            return;
        
        for (int i = len-1; n > 0 && m > 0; --i) {   //从结果数组的末尾开始
            if (nums2[n-1] > nums1[m-1]) {
                nums1[i] = nums2[n-1];
                --n;
            } else {
                nums1[i] = nums1[m-1];
                --m;
            }
        }
        while (n) {                                  //处理可能剩余的第二个数组
            nums1[n-1] = nums2[n-1];
            --n;
        }
    }
};

            Copiar el código provocará errores inexplicables. No es fácil de solucionar. Intente escribir el código usted mismo. Preste atención al aumento y disminución de variables temporales.

Supongo que te gusta

Origin blog.csdn.net/sy_123a/article/details/109231042
Recomendado
Clasificación