合并两个数组时出错

Question

我有两个排序数组 arr1 和 arr2，我正在尝试将这两个数组合并到另一个数组“输出”中。 但问题是，如果第一个数组用尽，即数组 arr1 被完全使用，那么“i”的值将大于数组的长度，因此在下一次迭代 arr[i] 时会出错。有什么办法可以避免这种情况？

    int i=0;
    int j=0;

    for(int k=0;k<output.length;k++) {      

            if(arr1[i]<arr2[j]) {
                output[k]=arr1[i];
                i++;
            }


            else{           
                   output[k]=arr2[j];
                   j++;         
            }       

    }

Answer 1

您需要检查数组是否已用尽：

int i=0;
int j=0;

for(int k=0;k<output.length;k++) {      
    if(i < arr1.length && j < arr2.length) { 
        // both arrays still have unused elements
        if (arr1[i]<arr2[j]) {
            output[k]=arr1[i];
            i++;
        } else {           
           output[k]=arr2[j];
           j++;         
        } 
    } else if (i < arr1.length) {
        // only the first array has unused elements
        output[k]=arr1[i];
        i++;   
    } else if (j < arr2.length) {
        // only the second array has unused elements
        output[k]=arr2[j];
        j++; 
    } 
}

Answer 2

我们可以利用这样一个事实，即不必检查 k 的值，因为输出数组的长度等于 arr1 和 arr2 的长度之和。一旦任一数组的元素耗尽，我们就退出主 for 循环（因此条件 i

int i=0;
int j=0;

for(int k=0;i<arr1.length && j < arr2.length ;k++) {      

        if(arr1[i]<arr2[j]) {
            output[k]=arr1[i];
            i++;
        }


        else{           
               output[k]=arr2[j];
               j++;         
        }       

}
if(i != arr1.length){
    for(; i < arr1.length; i++,k++)
    output[k]=arr1[i];
}
else{
    for(; j < arr2.length; j++,k++)
    output[k]=arr2[j];
}