leetcode 907. Sum of Subarray Minimums解題思路
阿新 • • 發佈:2018-11-02
題目大意是: 將各個連續子陣列的最小值累加,求個和值.
總體解題思路:
(1) 求出每個元素,在所有子陣列中成為最小值的數量.
(2)元素值 乘以 數量, 再累加求和.
詳細解題思路:
(1)求出各個元素的左邊界.
(2)求出各個元素的右邊界.
(3)在左邊界內取一個位置(C左 1) 乘以 在右邊界內 取一個位置(C右 1).
(4)注意相等的位置不能重複計算.
實現程式碼如下:
class Solution { public: int sumSubarrayMins(vector<int>& a) { int as=a.size(); vector<int> left(as,-1),right(as,as),st; for(int i=0;i<as;i++)//取左邊界 { while(!st.empty() && a[st.back()]>a[i]) { st.pop_back(); } if(!st.empty()) { left[i]=st.back(); } st.emplace_back(i); } st.clear(); for(int i=as-1;i>=0;i--)//取右邊界 { while(!st.empty() && a[st.back()]>=a[i])//相等只計算一次 { st.pop_back(); } if(!st.empty()) { right[i]=st.back(); } st.emplace_back(i); } long long rst=0; for(int i=0;i<as;i++)//每個元素的子陣列數量累加 { rst+=((i-left[i])*(right[i]-i)*a[i]); } return rst%1000000007; } };