LeetCode_面试题10.01_ 合并排序的数组(直接比较插入合并)

LeetCode_面试题10.01_ 合并排序的数组(直接比较插入合并)

题目描述:

给定两个排序后的数组 A 和 B,其中 A 的末端有足够的缓冲空间容纳 B。
编写一个方法,将 B 合并入 A 并排序。

初始化 A 和 B 的元素数量分别为 m 和 n。

示例:

输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6], n = 3

输出:
[1,2,2,3,5,6]

个人解答:

class Solution {
    public void merge(int[] A, int m, int[] B, int n) {
    
        //设置一个标记变量,减少循环比较次数
        int flag = 0;
        //按直接插入排序的方法将数组B的元素依次比较插入数组A中
        //对数组B作循环
        for (int i = 0; i < n; i ++) {
            //对数组A作循环            
            for (int j = flag; j < m + n; j ++) {                
            //通过比较寻找插入的下标                
            	if (B[i] < A[j] || (A[j] == 0 && j == m + i)) {                    
            		//将处于插入位置后的元素往后移动一个位置(包括待插入位置的元素)                    
            		for (int k = m + n - 1; k > j; k --){                        
                		A[k] = A[k - 1];                    
            		}                    
            		//将B中元素按序插入至A数组中                    
            		A[j] = B[i];                    
            		//将插入位置作为标记为flag,之前的元素就无需比较了                    
            		flag = j + 1;                    
            		break;                
            	}            
            }        
        }    
    }
}

执行结果分析:
LeetCode_面试题10.01_ 合并排序的数组(直接比较插入合并)_第1张图片

点击此处直接跳转至力扣官网该题页面

你可能感兴趣的:(笔记)