【Leetcode】2334. Subarray With Elements Greater Than Varying Threshold

题目地址：

https://leetcode.com/problems/subarray-with-elements-greater-than-varying-threshold/description/

给定一个长$n$正整数数组$a$，再给定一个正整数$t$，问是否存在一个$a$的非空长$k$的子数组使得该子数组每个数都大于$t/k$。如果存在，返回任意一个满足条件的子数组的长度；如果不存在，返回$-1$。

如果$a[l:r]$满足条件，设$\arg \min a[l:r]=k$，我们考虑最左边$l^{\prime }\le k$使得$a[l^{\prime }]\ge a[k]$和最右边的$r^{\prime }\ge k$使得$a[r^{\prime }]\ge a[k]$，那么$a[l^{\prime }:r^{\prime }]$一定也满足条件。所以我们只需要枚举形如$a[l^{\prime }:r^{\prime }]$的区间就行了，可以使用单调栈。代码如下：

class Solution {
 public:
  int validSubarraySize(vector<int>& a, int t) {
    stack<int> stk;
    int n = a.size();
    for (int i = 0; i < n; i++) {
      while (stk.size() && a[stk.top()] >= a[i]) {
        int idx = stk.top();
        stk.pop();
        int l = stk.size() ? stk.top() : -1;
        int len = i - l - 1;
        if (a[idx] > (double)t / len) return len;
      }
      stk.push(i);
    }
    while (stk.size()) {
      int idx = stk.top();
      stk.pop();
      int l = stk.size() ? stk.top() : -1;
      int len = n - l - 1;
      if (a[idx] > (double)t / len) return len;
    }
    return -1;
  }
};

时空复杂度$O(n)$。