用C++解决以下问题:给出一个长度为n的数组 a_1a_2a_3a_n 求一个字典序最小的1~n的排列P 使得原数组重新排列成a_P_1a_P_2a_P_3a_P_n后每一个前缀的平均数都大于等于 0 。输入样例:5-1 -2 -3 4 5输出样例:4 1 2 5 3
以下是使用C++解决该问题的代码:
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
bool compare(pair<int, double> a, pair<int, double> b) {
return a.second < b.second;
}
vector<int> getPermutation(vector<int> arr) {
int n = arr.size();
vector<pair<int, double>> prefixSums;
double sum = 0.0;
for (int i = 0; i < n; i++) {
sum += arr[i];
prefixSums.push_back(make_pair(i + 1, sum / (i + 1)));
}
sort(prefixSums.begin(), prefixSums.end(), compare);
vector<int> permutation;
for (int i = 0; i < n; i++) {
permutation.push_back(prefixSums[i].first);
}
return permutation;
}
int main() {
int n;
cin >> n;
vector<int> arr(n);
for (int i = 0; i < n; i++) {
cin >> arr[i];
}
vector<int> permutation = getPermutation(arr);
for (int i = 0; i < n; i++) {
cout << permutation[i] << " ";
}
cout << endl;
return 0;
}
该代码首先定义了一个compare函数,用于比较两个前缀平均数的大小。然后定义了getPermutation函数,该函数接受一个数组作为输入,并返回一个字典序最小的满足条件的排列。
在getPermutation函数中,首先计算了每个前缀的平均数,并将它们存储在一个prefixSums数组中。然后,使用sort函数对prefixSums数组进行排序,排序的依据是前缀平均数的大小。
最后,在main函数中,首先读取输入的数组,并调用getPermutation函数获得结果。最后,输出结果
原文地址: http://www.cveoy.top/t/topic/iQv2 著作权归作者所有。请勿转载和采集!