题目描述
放暑假了,小明决定到某旅游景点游玩,他在网上搜索到了各种价位的酒店(长度为 n 的数组 A),他的心理价位是 x 元,
请帮他筛选出 k 个最接近 x 元的酒店(n ≥ k > 0),并由低到高打印酒店的价格。
备注
- 酒店价格数组 A 和小明的心理价位 x 均为整型数据(0 < n, k, x < 10000)
- 优先选择价格最接近心理价位的酒店,若两家酒店距离心理价位差价相同,则选择价格较低的酒店。(比如 100 元和 300 元距离心理价位 200 元同样接近,此时选择 100 元)
- 酒店价格可能相同重复
输入描述
第一行:n, k, x
第二行:A[0] A[1] A[2]…A[n-1]
输出描述
从低到高打印筛选出的酒店价格
用例1
输入
10 5 6
1 2 3 4 5 6 7 8 9 10
Copy
输出
4 5 6 7 8
Copy
说明
数组长度 n = 10,筛选个数 k = 5,目标价位 x = 6
用例2
输入
10 4 6
10 9 8 7 6 5 4 3 2 1
Copy
输出
4 5 6 7
Copy
说明
数组长度 n = 10,筛选个数 k = 4,目标价位 x = 6
当4和8时距离 x 相同时,优先选择价格低的4
用例3
输入
6 3 1000
30 30 200 500 70 300
Copy
输出
200 300 500
#include <iostream>
#include <vector>
#include <algorithm>
#include <sstream>
using namespace std;
struct HotelInfo {
int price;
int distance;
};
string getResult(int n, int k, int x, const vector<int>& prices) {
vector<HotelInfo> tmp;
for (int p : prices) {
tmp.push_back({p, abs(p - x)});
}
sort(tmp.begin(), tmp.end(), [](const HotelInfo& a, const HotelInfo& b) {
if (a.distance!= b.distance)
return a.distance < b.distance;
else
return a.price < b.price;
});
vector<int> ans;
for (int i = 0; i < k && i < tmp.size(); i++) {
ans.push_back(tmp[i].price);
}
sort(ans.begin(), ans.end());
stringstream ss;
for (int an : ans) {
ss << an << " ";
}
return ss.str();
}
int main() {
int n, k, x;
cin >> n >> k >> x;
vector<int> prices(n);
for (int i = 0; i < n; i++) {
cin >> prices[i];
}
cout << getResult(n, k, x, prices) << endl;
return 0;
}