【Leetcode】1124. Longest Well-Performing Interval

题目地址：

https://leetcode.com/problems/longest-well-performing-interval/

给定一个长$n$的数组$A$，求其最长的子数组，使得该子数组里的大于$8$的数的个数比小于等于$8$的个数要多。返回子数组长度。

题目等价于为给定一个只含$1,-1$的数组，求其和大于等于$1$的最长子数组的长度。这可以用前缀和 + 单调栈 + 二分来做（具体做法是开一个严格单调下降的栈，设当前前缀和为$x$，栈顶为$y$，显然当$x\le y$的时候，$y$是不会作为子数组左端点的左边那个数的）。本题比较特殊，当要寻找小于等于$x-1$的数的最大前缀和的时候，不需要用二分，只需要哈希表就可以了，原因是栈里如果有小于等于$x-1$的前缀和，那就一定恰好有$x-1$，否则的话前缀和是不可能从某个小于$x-1$的数不连续地跳到$x$的，因为$A$只含$1,-1$。代码如下：

class Solution {
 public:
  int longestWPI(vector<int>& A) {
    unordered_map<int, int> mp;
    int res = 0;
    for (int i = 0, sum = 0; i < A.size(); i++) {
      sum += A[i] > 8 ? 1 : -1;
      if (sum > 0) res = i + 1;
      else {
        if (mp.count(sum - 1)) res = max(res, i - mp[sum - 1]);
        if (!mp.count(sum)) mp[sum] = i;
      }
    }

    return res;
  }
};

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