1. 程式人生 > >[Swift-2019力扣杯春季決賽]4. 有效子數組的數目

[Swift-2019力扣杯春季決賽]4. 有效子數組的數目

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. 1 <= A.length <= 50000
  2. 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.count
7 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 += 1
17 st[t] = i 18 ans += ne[i] - i 19 } 20 return ans 21 } 22 }

[Swift-2019力扣杯春季決賽]4. 有效子數組的數目