使用c++编写一个程序,利用priority_queue(优先队列),求出一个无序整数序列中第k小的元素
时间: 2024-10-14 12:03:38 浏览: 33
STL.rar_C++ STL_C++ STL_STL_STL c++_STL教程
在C++中,我们可以使用`priority_queue`数据结构,配合一个辅助数组来解决这个问题。`priority_queue`本质上是一个最小堆,所以我们将优先队列用于存储序列中的元素,并不断插入,直到队列大小达到`k`。然后,当我们插入第`k`个元素时,由于它是当前最小的`k`个元素之一,所以可以直接返回它,而不需要完全遍历整个序列。
以下是一个简单的示例程序:
```cpp
#include <queue>
#include <vector>
int findKthSmallest(int arr[], int n, int k) {
std::priority_queue<int, std::vector<int>, std::greater<int>> pq;
// 将前k个元素入队
for (int i = 0; i < k; ++i) {
pq.push(arr[i]);
}
// 遍历剩余元素,如果新的元素小于队首,则替换队首并调整队列大小
for (int i = k; i < n; ++i) {
if (pq.top() > arr[i]) {
pq.pop();
pq.push(arr[i]);
}
}
// 返回队首元素,即第k小的元素
return pq.top();
}
int main() {
int arr[] = {5, 3, 9, 7, 1};
int n = sizeof(arr) / sizeof(arr[0]);
int k = 3;
int kth_smallest = findKthSmallest(arr, n, k);
std::cout << "The " << k << "th smallest element is: " << kth_smallest << std::endl;
return 0;
}
```
在这个程序中,我们首先初始化一个优先队列,将前`k`个元素放入。然后遍历剩下的元素,如果遇到比队首元素还小的新元素,就将其替换掉队首并将队列大小减一。遍历结束后,队头的元素就是第`k`小的元素。
阅读全文