資料結構
網路
關係資料庫管理系統 (RDBMS)
作業系統
Java
MS Excel
iOS
HTML
CSS
Android
Python
C語言程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP
物理學
化學
生物學
數學
英語
經濟學
心理學
社會學
服裝設計
法學在C++中刪除給定元素後查詢k個最小數
在這個問題中，我們得到一個大小為n的陣列arr[]，大小為m的陣列del[]和一個整數k。我們的任務是*在刪除給定元素後查詢k個最小數*。
我們需要列印在刪除del[]陣列中存在的所有元素後找到的arr[]陣列的前k個最小元素。如果陣列中存在兩個例項，則刪除第一個例項。
讓我們來看一個例子來理解這個問題：
Input : arr[] = {3, 5, 1, 7, 9, 2}, del[] = {1, 9, 3}, k = 2
Output : 2, 5**解釋** −
Array arr[] after deleting the elements : {5, 7, 2}
2 minimum elements are 2, 5.解決方案方法
解決這個問題的一個簡單方法是刪除arr[]中所有在del[]中存在的元素。然後按升序對陣列進行排序，並列印陣列的前k個元素。
示例
程式說明我們解決方案的工作原理
#include <bits/stdc++.h>
using namespace std;
void findKminElementDelArray(int arr[], int n, int del[], int m, int k){
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
if(arr[j] == del[i]){
arr[j] = INT_MAX;
break;
}
}
}
sort(arr, arr + n);
for (int i = 0; i < k; ++i) {
cout<<arr[i]<<" ";
}
}
int main(){
int array[] = { 3, 5, 1, 7, 9, 2 };
int m = sizeof(array) / sizeof(array[0]);
int del[] = { 1, 9, 3 };
int n = sizeof(del) / sizeof(del[0]);
int k = 2;
cout<<k<<" smallest numbers after deleting the elements are ";
findKminElementDelArray(array, m, del, n, k);
return 0;
}輸出
2 smallest numbers after deleting the elements are 2 5
另一種方法
解決這個問題的另一種方法是使用雜湊表和堆。我們將建立一個最小堆和一個雜湊表。雜湊表將包含del[]陣列的所有元素。然後將不在雜湊表中的arr[]陣列的元素插入到最小堆中。從堆中彈出k個元素，然後打印出來。
示例
程式說明我們解決方案的工作原理
#include <bits/stdc++.h>
using namespace std;
void findKminElementDelArray(int arr[], int n, int del[], int m, int k){
unordered_map<int, int> deleteElement;
for (int i = 0; i < m; ++i) {
deleteElement[del[i]]++;
}
priority_queue<int, vector<int>, greater<int> > minHeap;
for (int i = 0; i < n; ++i) {
if (deleteElement.find(arr[i]) != deleteElement.end()) {
deleteElement[arr[i]]--;
if (deleteElement[arr[i]] == 0) deleteElement.erase(arr[i]);
}
else
minHeap.push(arr[i]);
}
for (int i = 0; i < k; ++i) {
cout<<minHeap.top()<<" ";
minHeap.pop();
}
}
int main(){
int array[] = { 3, 5, 1, 7, 9, 2 };
int m = sizeof(array) / sizeof(array[0]);
int del[] = { 1, 9, 3 }; int n = sizeof(del) / sizeof(del[0]);
int k = 2;
cout<<k<<" smallest numbers after deleting the elements are ";
findKminElementDelArray(array, m, del, n, k);
return 0;
}輸出
2 smallest numbers after deleting the elements are 2 5
更新於：2022年1月28日
瀏覽量：133
廣告