Commit de51403

authored

Create 3676.Count-Bowl-Subarrays.cpp

1 parent 0899ed9 commit de51403Copy full SHA for de51403

File tree

+28

-0

lines changed

+28

-0

lines changed

Lines changed: 28 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,28 @@`
	`1`	`+class Solution {`
	`2`	`+public:`
	`3`	`+ long long bowlSubarrays(vector<int>& nums) {`
	`4`	`+ int n = nums.size();`
	`5`	`+ vector<int>nextGreater(n, -1);`
	`6`	`+ vector<int>prevGreater(n, -1);`
	`7`	`+ vector<int>st;`
	`8`	`+ for (int i=0; i<n; i++) {`
	`9`	`+ while (!st.empty() && nums[st.back()]<nums[i]) st.pop_back();`
	`10`	`+ if (!st.empty()) prevGreater[i] = st.back();`
	`11`	`+ st.push_back(i);`
	`12`	`+ }`
	`13`	`+ st.clear();`
	`14`	`+`
	`15`	`+ for (int i=n-1; i>=0; i--) {`
	`16`	`+ while (!st.empty() && nums[st.back()]<nums[i]) st.pop_back();`
	`17`	`+ if (!st.empty()) nextGreater[i] = st.back();`
	`18`	`+ st.push_back(i);`
	`19`	`+ }`
	`20`	`+`
	`21`	`+ long long ret = 0;`
	`22`	`+ for (int i=0; i<n; i++) {`
	`23`	`+ if (prevGreater[i]!=-1 && i-prevGreater[i]>=2) ret++;`
	`24`	`+ if (nextGreater[i]!=-1 && nextGreater[i]-i>=2) ret++;`
	`25`	`+ }`
	`26`	`+ return ret;`
	`27`	`+ }`
	`28`	`+};`

Comments

(0)