Neat Tree

題意：

n個數，每個區間的貢獻為區間內最大值減最小值，問這個n個數的總貢獻是多少？也就是n所能組成的所有區間的貢獻值之和

題解：

我們可以用單調棧來做
第i個數對答案的貢獻值為=h[i] * 作為最大值出現的次數-h[i] * 作為最小值出現的次數
求每個點貢獻值的過程可以用單調棧來維護
例如求最大值的次數：
對于第i點，找到左端第一個比h[i]大的數，記其位置為L，找到右端第一個比h[i]大的數，記其位置為R，(默認第0個位置最小，第n+1個位置最大)則區間(L,R)區間內包含點i的子區間個數就是(i-L) * (R-i)
答案的貢獻就是h[i] * (i-L) * (R-i)

ll sum=0;stack<int>stk;for(int i=1;i<=n;i++)cin>>h[i];for(int i=1;i<+n;i++){while(!stk.empty()&&h[stk.top()]>=h[i])stk.pop();//記錄左端第一個比h[i]小的index if(stk.empty())l[i]=1;//如果棧內無元素，取左端 else l[i]=stk.pop()+1;//stk.pop()是第一個比h[i]小的坐標，加一則是恰好不小于h[i]的 stk.push(i); }while(!stk.empty())stk.pop();//清空棧

代碼：

#include<cstdio> #include<queue> #include<stack> #include<iostream> using namespace std; const int maxn=1e6+10; typedef long long ll; int l[maxn], r[maxn]; ll h[maxn];int main(){int n;while(~scanf("%d", &n)){for(int i=1; i<=n; i++)scanf("%lld", &h[i]);stack<int> s;for(int i=1; i<=n; i++){while(!s.empty() && h[i]>=h[s.top()]) s.pop();//注意等號問題，因為區間不要重復。if(s.empty()) l[i]=0;else l[i]=s.top();s.push(i);}//--------------計算每個區間的最大值之和 while(!s.empty()) s.pop();for(int i=n; i>=1; i--){while(!s.empty() && h[i]>h[s.top()]) s.pop();if(s.empty()) r[i]=n+1;else r[i]=s.top();s.push(i);}ll maxx=0;for(int i=1; i<=n; i++)maxx+=h[i]*(i-l[i])*(r[i]-i);//----------計算每個區間的最小值之和 while(!s.empty()) s.pop();for(int i=1; i<=n; i++){while(!s.empty() && h[i]<=h[s.top()]) s.pop();if(s.empty()) l[i]=0;else l[i]=s.top();s.push(i);}while(!s.empty()) s.pop();for(int i=n; i>=1; i--){while(!s.empty() && h[i]<h[s.top()]) s.pop();if(s.empty()) r[i]=n+1;else r[i]=s.top();s.push(i);}ll minn=0;for(int i=1; i<=n; i++)minn+=h[i]*(i-l[i])*(r[i]-i);printf("%lld\n", maxx-minn);}return 0; } 創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的Neat Tree的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。