[Swift-2019力扣杯春季決賽]4. 有效子數組的數目
阿新 • • 發佈:2019-04-21
bin 返回 turn pan 解釋 while stride pre run
給定一個整數數組 A
,返回滿足下面條件的 非空、連續 子數組的數目:
子數組中,最左側的元素不大於其他元素。
示例 1:
輸入:[1,4,2,5,3] 輸出:11 解釋:有 11 個有效子數組,分別是:[1],[4],[2],[5],[3],[1,4],[2,5],[1,4,2],[2,5,3],[1,4,2,5],[1,4,2,5,3] 。
示例 2:
輸入:[3,2,1] 輸出:3 解釋:有 3 個有效子數組,分別是:[3],[2],[1] 。
示例 3:
輸入:[2,2,2] 輸出:6 解釋:有 6 個有效子數組,分別為是:[2],[2],[2],[2,2],[2,2],[2,2,2] 。
提示:
1 <= A.length <= 50000
0 <= A[i] <= 100000
848 ms
1 class Solution { 2 var ne:[Int] = [Int](repeating:0,count:50005) 3 var st:[Int] = [Int](repeating:0,count:50005) 4 var t:Int = 0 5 func validSubarrays(_ nums: [Int]) -> Int { 6 var n:Int = nums.count7 var ans:Int = 0 8 st[0] = n 9 for i in stride(from:(n - 1),through:0,by:-1) 10 { 11 while((t != 0) && nums[st[t]] >= nums[i]) 12 { 13 t -= 1 14 } 15 ne[i] = st[t] 16 t += 117 st[t] = i 18 ans += ne[i] - i 19 } 20 return ans 21 } 22 }
[Swift-2019力扣杯春季決賽]4. 有效子數組的數目