合并K个有序数组(假设每个数组长度相等)

直接上代码,看注释

#include <iostream>
using namespace std;

struct HeapNode {
	int val;//num value
	int i;//array index
	int j;//num index
};

//length of each array
static const int numLen = 3;

//adjust minheap
void MinHeapFixDown(HeapNode data[], int i, const int len) {
	HeapNode tmp = data[i];
	int j = 2*i+1;
	while (j < len) {
		if (j+1 < len && data[j+1].val < data[j].val)
			++j;
		if (data[j].val >= tmp.val)
			break;
		data[i] = data[j];
		i = j;
		j = 2*i+1;
	}
	data[i] = tmp;
}

//Create Min Heap
void MakeMinHeap(HeapNode data[], const int len) {
	for (int i = len/2-1; i >= 0; --i)
		MinHeapFixDown(data, i, len);
}

//Merge Array
void MergeArray(int nums[][numLen], const int k, int res[]) {
	//temp array used to store nodes
	HeapNode tmp[k];
	//initialize the tmp by the first element of each array
	for (int i = 0; i < k; ++i){
		tmp[i].val = nums[i][0];
		tmp[i].i = i;
		tmp[i].j = 0;
	}
	//Create the heap
	MakeMinHeap(tmp, k);
	int count = 0; //count the number of arrays been merged
	int index = 0; //result array index
	//until all the arrays been merged
	while (count != k) {
		//the first element of the array is the minimum
		res[index++] = tmp[0].val;
		//if not walk to the end of one array
		if (tmp[0].j != numLen-1) {
			//update the the node into the next node in the ith array
			tmp[0].j = tmp[0].j+1;
			tmp[0].val = nums[tmp[0].i][tmp[0].j];
			//adjust the heap
			MinHeapFixDown(tmp, 0, k-count);
		}
		//if walk to the end of one array
		else {
			//count plus one
			++count;
			//assign the first node with the last node
			tmp[0] = tmp[k-count];
			//adjust the heap
			//Note::len is k-count
			MinHeapFixDown(tmp, 0, k-count);
		}
	}
}

int main() {
	const int row = 6;
	int nums[row][numLen] = {{22,112,210},{2,5,12},{6,9,13},{7,8,10},{3,11,28},{13,14,15}};
	int res[row*numLen];
	MergeArray(nums, row, res);
	for (int i = 0 ; i < row*numLen; ++i)
		cout<<res[i]<<endl;
	return 0;
}


### 回答1: 可以使用双指针法和额外空间来合并两个有序数组。具体实现如下: ```java public void merge(int[] nums1, int m, int[] nums2, int n) { int[] merged = new int[m + n]; int i = 0, j = 0, k = 0; while (i < m && j < n) { if (nums1[i] < nums2[j]) { merged[k++] = nums1[i++]; } else { merged[k++] = nums2[j++]; } } while (i < m) { merged[k++] = nums1[i++]; } while (j < n) { merged[k++] = nums2[j++]; } System.arraycopy(merged, 0, nums1, 0, m + n); } ``` 其中,`i`、`j`、`k` 分别代表 `nums1`、`nums2`、`merged` 数组的下标,`m`、`n` 分别代表 `nums1`、`nums2` 的长度。在循环中,比较 `nums1[i]` 和 `nums2[j]` 的大小,将小的数放入 `merged` 数组中,并将对应的指针向后移动一位。最后,将剩余的元素依次放入 `merged` 数组中,并将 `merged` 数组复制到 `nums1` 数组中。时间复杂度为 $O(m+n)$,空间复杂度为 $O(m+n)$。 ### 回答2: 在Java中合并两个有序数组可以使用双指针的方式进行。首先,我们需要创建一个新的数组来存放合并后的结果。然后,我们定义两个指针分别指向两个有序数组的起始位置。接下来,我们开始比较两个指针所指向的元素的大小。 1. 如果第一个数组的当前元素小于第二个数组的当前元素,则将第一个数组的元素放入新数组,并将第一个数组的指针后移一位。 2. 如果第一个数组的当前元素大于第二个数组的当前元素,则将第二个数组的元素放入新数组,并将第二个数组的指针后移一位。 3. 如果两个数组的当前元素相等,则将其中一个元素放入新数组,并将两个数组的指针都后移一位。 重复以上步骤,直到其中一个数组的指针超出了数组的范围。然后,将剩余的元素依次放入新数组中。最后,返回新数组即可。 以下是一个示例代码实现: ```java public static int[] mergeArrays(int[] arr1, int[] arr2) { int[] merged = new int[arr1.length + arr2.length]; int i = 0, j = 0, k = 0; while (i < arr1.length && j < arr2.length) { if (arr1[i] < arr2[j]) { merged[k] = arr1[i]; i++; } else { merged[k] = arr2[j]; j++; } k++; } while (i < arr1.length) { merged[k] = arr1[i]; i++; k++; } while (j < arr2.length) { merged[k] = arr2[j]; j++; k++; } return merged; } ``` 通过调用`mergeArrays`方法,可以将两个有序数组合并为一个有序数组。 ### 回答3: 合并两个有序数组是指将两个已经按照升序排列的数组合并为一个按升序排列的数组。在Java中,我们可以使用双指针的方法来解决这个问题。 具体步骤如下: 1. 创建一个新数组,用来保存合并后的结果。 2. 创建两个指针,分别指向两个数组的开头。 3. 循环比较两个指针所指向的元素,将较小的元素放入新数组中,并将对应的指针向后移动一位。 4. 当其中一个指针已经遍历完了一个数组时,将另一个数组中剩下的元素依次放入新数组中。 5. 返回新数组即为合并后的结果。 以下是使用Java代码实现的例子: ```java public int[] mergeArrays(int[] nums1, int[] nums2) { int len1 = nums1.length; int len2 = nums2.length; int[] result = new int[len1 + len2]; int i = 0, j = 0, k = 0; while (i < len1 && j < len2) { if (nums1[i] <= nums2[j]) { result[k++] = nums1[i++]; } else { result[k++] = nums2[j++]; } } while (i < len1) { result[k++] = nums1[i++]; } while (j < len2) { result[k++] = nums2[j++]; } return result; } ``` 以上代码中,通过两个while循环分别处理了某个数组已经遍历完的情况,最后返回合并后的新数组result即可。 希望能帮助到你!
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值