Codility上的问题 (17) PI 2012

时间:2023-03-09 23:39:02
Codility上的问题 (17) PI 2012

这个题比较简单,给定一个整数数组,对每个元素,求出和它最近比它大的数的距离(下标绝对值),如果没有比它大的数,认为距离是0。

数组元素个数 N [0..50000],数组元素范围[-10^9, +10^9]。

要求复杂度 时间 空间 都是O(N)。

分析:

这个题比较简单吧。跟直方图最大矩形差不多,类似于求左右边界。求左边界的话记住,有这个数在的话,比它更早的并且比它小的数都没有意义(因为有这个数存在,而它又很大,右边的数往左找的话会先选择这个数)。于是栈内元素是单调递减的,求右边界类似。

代码:

// you can also use includes, for example:
// #include <algorithm> vector<int> solution(const vector<int> &A) {
// write your code here...
int n = A.size(),i;
vector<int> a,r;
r.resize(n);
for (i = 0; i < n; ++i) {
while ((!a.empty()) && (A[a.back()] <= A[i])) {
a.pop_back();
}
r[i] = a.empty()?n:(i - a.back());
a.push_back(i);
}
a.clear();
for (i = n - 1; i >= 0; --i) {
while ((!a.empty()) && (A[a.back()] <= A[i])) {
a.pop_back();
}
if (!a.empty()) {
r[i] = min(r[i], a.back() - i);
}
else if (r[i] == n) {
r[i] = 0;
}
a.push_back(i);
}
return r; }