贝壳极差题目O(n)解法

时间限制:C/C++语言 1000MS;其他语言 3000MS
内存限制: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;

}

#笔试题目##贝壳找房#
全部评论
大佬解释下区间总数为什么这么算啊,没看懂…
点赞 回复 分享
发布于 2018-08-19 13:08
大佬 厉害了 
点赞 回复 分享
发布于 2018-08-18 21:30
暴力过了。。。
点赞 回复 分享
发布于 2018-08-18 21:35
极差是啥,修路那个?
点赞 回复 分享
发布于 2018-08-18 21:47
竟然是上次头条一面的第一个代码题
点赞 回复 分享
发布于 2018-08-18 21:52
这是啥题,我为啥没做,一共几个题😱
点赞 回复 分享
发布于 2018-08-18 21:54
描述有点抽象。。。
点赞 回复 分享
发布于 2018-08-18 23:58
这个缩进看得我难受,自己对着写了遍,测试样例都没过,楼主看看哪里错了。 typedef long long LL; int main() {     int n;     cin >> n;     if (n <= 1)     {         cout << 0 << endl;         return 0;     }     vector<LL> a(n);     for (int i = 0; i < n; i++)         cin >> a[i];     stack<pair<LL, 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(make_pair(a[i], i));     }     s = stack<pair<LL, 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 });     }     LL res = 0;     for (int i = 0; i < n; i++)     {         res -= a[i] * (i - c[i]) * (d[i] - i);     }     s = stack<pair<LL, 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<LL, 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] - 1);     }     cout << res << endl;     return 0; }
点赞 回复 分享
发布于 2018-08-19 16:23
这个题目在哪里能看到?
点赞 回复 分享
发布于 2018-09-01 20:44

相关推荐

10-16 22:56
门头沟学院 C++
1234567800:歌尔今年给211开14-15k吗,我本地人连面试都不给😂
点赞 评论 收藏
分享
评论
点赞
31
分享
牛客网
牛客企业服务