Trong hướng dẫn này, chúng ta sẽ viết một chương trình tìm phần tử thứ k từ mảng đã hợp nhất của hai mảng đã sắp xếp.
Hãy xem các bước để giải quyết vấn đề.
- Khởi tạo hai mảng đã sắp xếp.
- Khởi tạo một mảng trống có kích thước m + n.
- Hợp nhất hai mảng thành mảng mới.
- Trả về phần tử k - 1 từ mảng đã hợp nhất.
Ví dụ
Hãy xem mã.
#include <iostream>
using namespace std;
int findKthElement(int arr_one[], int arr_two[], int m, int n, int k) {
int sorted_arr[m + n];
int i = 0, j = 0, index = 0;
while (i < m && j < n) {
if (arr_one[i] < arr_two[j]) {
sorted_arr[index++] = arr_one[i++];
}else {
sorted_arr[index++] = arr_two[j++];
}
}
while (i < m) {
sorted_arr[index++] = arr_one[i++];
}
while (j < n) {
sorted_arr[index++] = arr_two[j++];
}
return sorted_arr[k - 1];
}
int main() {
int arr_one[5] = {1, 3, 5, 7, 9}, arr_two[5] = {2, 4, 6, 8, 10};
int k = 7;
cout << findKthElement(arr_one, arr_two, 5, 4, k) << endl;
return 0;
} Đầu ra
Nếu bạn chạy đoạn mã trên, thì bạn sẽ nhận được kết quả sau.
7
Kết luận
Nếu bạn có bất kỳ câu hỏi nào trong hướng dẫn, hãy đề cập đến chúng trong phần bình luận.