贝壳极差题目O（n）解法
时间限制：C/C++语言 1000MS；其他语言 3000MS
内存限制：C/C++语言 65536KB；其他语言 589824KB
题目描述：
给定一个长度为N的序列A1到AN，求所有区间[L，R]（1≤L≤R≤N）的极差之和，其中区间[L，R]的极差定义为AL到AR中的最大值与最小值之差。
内存限制：C/C++语言 65536KB；其他语言 589824KB
题目描述：
给定一个长度为N的序列A1到AN，求所有区间[L，R]（1≤L≤R≤N）的极差之和，其中区间[L，R]的极差定义为AL到AR中的最大值与最小值之差。
贝壳极差题目O（n）解法，花了很长时间才做出来，不要跟我说暴力就过了。。。。。。
思路：先算减去所有区间最小值，再算加上所有区间最大值。用单调栈可以算出，比某一个数大的最近的数，比如i位置的数，分别对应比他大的最近的数的位置是ci和di，那么所有的区间总数就是（i-ci）*（di-i），然后用ai乘上它，再从res里面减去。最大值得思路一样。
using namespace std;
int main() {
int n;
cin >> n;
if (n <= 1) {
cout << 0;
return 0;
}
vector<long long>a(n);
for (int i = 0; i < n; i++)
cin >> a[i];
stack<pair<int, int>> s;
vector<int> c(n), d(n);
for (int i = 0; i < n; i++) {
while (!s.empty() && s.top().first > a[i])
s.pop();
if (s.empty())
c[i] = -1;
else
c[i] = s.top().second;
s.push({ a[i], i });
}
s = stack<pair<int, int>>();
for (int i = n - 1; i >= 0; i--) {
while (!s.empty() && s.top().first >= a[i])
s.pop();
if (s.empty())
d[i] = n;
else
d[i] = s.top().second;
s.push({ a[i], i });
}
long long res = 0;
for (int i = 0; i < n; i++) {
res -= a[i] * (i - c[i]) * (d[i] - i);
}
s = stack<pair<int, int>>();
for (int i = 0; i < n; i++) {
while (!s.empty() && s.top().first <= a[i])
s.pop();
if (s.empty())
c[i] = -1;
else
c[i] = s.top().second;
s.push({ a[i], i });
}
s = stack<pair<int, int>>();
for (int i = n - 1; i >= 0; i--) {
while (!s.empty() && s.top().first < a[i])
s.pop();
if (s.empty())
d[i] = n;
else
d[i] = s.top().second;
s.push({ a[i], i });
}
for (int i = 0; i < n; i++) {
res += a[i] * (i - c[i]) * (d[i] - i);
}
cout << res;
return 0;
}
int main() {
int n;
cin >> n;
if (n <= 1) {
cout << 0;
return 0;
}
vector<long long>a(n);
for (int i = 0; i < n; i++)
cin >> a[i];
stack<pair<int, int>> s;
vector<int> c(n), d(n);
for (int i = 0; i < n; i++) {
while (!s.empty() && s.top().first > a[i])
s.pop();
if (s.empty())
c[i] = -1;
else
c[i] = s.top().second;
s.push({ a[i], i });
}
s = stack<pair<int, int>>();
for (int i = n - 1; i >= 0; i--) {
while (!s.empty() && s.top().first >= a[i])
s.pop();
if (s.empty())
d[i] = n;
else
d[i] = s.top().second;
s.push({ a[i], i });
}
long long res = 0;
for (int i = 0; i < n; i++) {
res -= a[i] * (i - c[i]) * (d[i] - i);
}
s = stack<pair<int, int>>();
for (int i = 0; i < n; i++) {
while (!s.empty() && s.top().first <= a[i])
s.pop();
if (s.empty())
c[i] = -1;
else
c[i] = s.top().second;
s.push({ a[i], i });
}
s = stack<pair<int, int>>();
for (int i = n - 1; i >= 0; i--) {
while (!s.empty() && s.top().first < a[i])
s.pop();
if (s.empty())
d[i] = n;
else
d[i] = s.top().second;
s.push({ a[i], i });
}
for (int i = 0; i < n; i++) {
res += a[i] * (i - c[i]) * (d[i] - i);
}
cout << res;
return 0;
}
#笔试题目##贝壳找房#
投递腾讯等公司10个岗位 >
