CodeFroces 817D Imbalanced Array(单调栈)

来源:互联网 发布:php上传文件原理 编辑:程序博客网 时间:2024/06/01 07:58

题意:就是求出所有区间最大值减去最小值的和。

解法:学了一波单调栈。

用单调栈正反跑一遍得到每个数字若是最小值的时候能在的区间的范围。

正反跑一遍得到每个数字若是最大值的时候能在区间的范围。

然后求一求每个数字在多少个区间里面,求出贡献度,然后即可得出答案。

代码如下:

#include<iostream>#include<cstdio>#include<vector>#include<queue>#include<utility>#include<stack>#include<algorithm>#include<cstring>#include<string>using namespace std;typedef pair<int, int> pii;const int maxn = 1e6 + 5;long long a[maxn];int n;stack <pii> sta;struct node {long long left, right;}Min[maxn], Max[maxn];int main() {#ifndef ONLINE_JUDGEfreopen("in.txt", "r", stdin);//    freopen("out.txt", "w", stdout);#endifscanf("%d", &n);for(int i = 0; i < n; i++) {scanf("%I64d", &a[i]);}sta.push(pii(a[0], 0));//选取小的 Min[0].left = 0;for(int i = 1; i < n; i++) {int tmp = i;while(!sta.empty() && sta.top().first >= a[i]) {tmp = sta.top().second;sta.pop();}if(tmp != i)Min[i].left = Min[tmp].left;elseMin[i].left = i;sta.push(pii(a[i], i));}while(!sta.empty())sta.pop();sta.push(pii(a[n - 1], n - 1));Min[n - 1].right = n - 1;for(int i = n - 2; i >= 0; i--) {int tmp = i;while(!sta.empty() && sta.top().first > a[i]) {tmp = sta.top().second;sta.pop();}if(tmp != i)Min[i].right = Min[tmp].right;elseMin[i].right = i;sta.push(pii(a[i], i));}while(!sta.empty())//选取大的 sta.pop();sta.push(pii(a[0], 0));Max[0].left = 0;for(int i = 1; i < n; i++) {int tmp = i;while(!sta.empty() && sta.top().first <= a[i]) {tmp = sta.top().second;sta.pop();}if(tmp != i)Max[i].left = Max[tmp].left;elseMax[i].left = i;sta.push(pii(a[i], i));}while(!sta.empty())sta.pop();sta.push(pii(a[n - 1], n - 1));Max[n - 1].right = n - 1;for(int i = n - 2; i >= 0; i--) {int tmp = i;while(!sta.empty() && sta.top().first < a[i]) {tmp = sta.top().second;sta.pop();}if(tmp != i)Max[i].right = Max[tmp].right;elseMax[i].right = i;sta.push(pii(a[i], i));}long long sum1 = 0, sum2 = 0;//求出答案 for(int i = 0; i < n; i++) {sum1 += ((Min[i].right - i) + (i - Min[i].left) + (Min[i].right - i) * (i - Min[i].left)) * a[i];}for(int i = 0; i < n; i++) {sum2 += ((Max[i].right - i) + (i - Max[i].left) + (Max[i].right - i) * (i - Max[i].left)) * a[i];}printf("%I64d\n", sum2 - sum1);return 0;}


原创粉丝点击