Computer >> Máy Tính >  >> Lập trình >> C ++

Số lượng mảng con hợp lệ trong C ++

Giả sử chúng ta có một mảng A gồm các số nguyên, chúng ta phải tìm số mảng con liên tục không rỗng thỏa mãn điều kiện này:Phần tử ngoài cùng bên trái của mảng con không lớn hơn các phần tử khác trong mảng con.

Vì vậy, nếu đầu vào giống như [1,4,2,5,3], thì đầu ra sẽ là 11, vì có 11 mảng con hợp lệ, chúng giống như [1], [4], [2], [5 ], [3], [1,4], [2,5], [1,4,2], [2,5,3], [1,4,2,5], [1,4,2 , 5,3].

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

  • ret:=0

  • n:=kích thước của nums

  • Xác định một ngăn xếp

  • để khởi tạo i:=0, khi tôi

    • x:=nums [i]

    • while (not st là trống và x

      • xóa phần tử khỏi st

    • chèn x vào st

    • ret:=kích thước của st + ret

  • trả lại ret

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

Ví dụ

#include <bits/stdc++.h>
using namespace std;
class Solution {
   public:
   int validSubarrays(vector<int>& nums) {
      int ret = 0;
      int n = nums.size();
      stack <int> st;
      for(int i = 0; i < nums.size(); i++){
         int x = nums[i];
         while(!st.empty() && x < st.top()) st.pop();
         st.push(x);
         ret += st.size();
      }
      return ret;
   }
};
main(){
   Solution ob;
   vector<int> v = {1,4,2,5,3};
   cout << (ob.validSubarrays(v));
}

Đầu vào

{1,4,2,5,3}

Đầu ra

11