文章目录
例题——2488. 统计中位数为 K 的子数组⭐
https://leetcode.cn/problems/count-subarrays-with-median-k/description/
提示:
n == nums.length
1 <= n <= 10^5
1 <= nums[i], k <= n
nums 中的整数互不相同
【套路】子数组统计问题常用技巧:等价转换
https://leetcode.cn/problems/count-subarrays-with-median-k/solutions/1993439/deng-jie-zhuan-huan-pythonjavacgo-by-end-5w11/
注意,每个数字各不相同,整个数组中只有一个 k。
class Solution {
public int countSubarrays(int[] nums, int k) {
int pos = 0, n = nums.length;
// 求出 k 的 pos
while (nums[pos] != k) ++pos;
Map<Integer, Integer> cnt = new HashMap<>();
cnt.put(0, 1);
for (int i = pos - 1, x = 0; i >= 0; --i) {
x += nums[i] < k? 1: -1;
cnt.merge(x, 1, Integer::sum);
}
int ans = cnt.get(0) + cnt.getOrDefault(-1, 0);
for (int i = pos + 1, x = 0; i < n; ++i) {
x += nums[i] > k? 1: -1;
// x 和 x - 1 对应 奇数长度和偶数长度
ans += cnt.getOrDefault(x, 0) + cnt.getOrDefault(x - 1, 0);
}
return ans;
}
}
本质上是「中心前后缀 + 哈希表计数」
核心在于「个数」,能得到一个和个数有关的数学式子,就可以转换成 1(或者 -1),毕竟在讨论个数的时候,每个元素的贡献就是 1 了。
相似题目列表
面试题 17.05. 字母与数字
https://leetcode.cn/problems/find-longest-subarray-lcci/description/
前缀和设置为 字母与数字数量之差,相同前缀和的一对可以组成一个子字符串。
class Solution {
public String[] findLongestSubarray(String[] array) {
int st = 0, ml = 0;
int n = array.length, d = 0;
Map<Integer, Integer> m = new HashMap<>();
m.put(0, 0);
for (int i = 0; i < n; ++i) {
char ch = array[i].charAt(0);
if (ch >= '0' && ch <= '9') d++;
else d--;
if (m.containsKey(d)) {
int j = m.get(d);
if (i - j + 1 > ml) {
st = j;
ml = i - j + 1;
}
} else m.put(d, i + 1);
}
String[] ans = new String[ml];
System.arraycopy(array, st, ans, 0, ml);
return ans;
}
}
525. 连续数组
https://leetcode.cn/problems/contiguous-array/description/
跟上面那道题几乎一模一样。
class Solution {
public int findMaxLength(int[] nums) {
int d = 0, ans = 0;
Map<Integer, Integer> m = new HashMap<>();
m.put(0, 0);
for (int i = 0; i < nums.length; ++i) {
if (nums[i] == 1) d++;
else d--;
if (m.containsKey(d)) {
int id = m.get(d);
if (i - id + 1 > ans) ans = i - id + 1;
} else m.put(d, i + 1);
}
return ans;
}
}
1124. 表现良好的最长时间段
https://leetcode.cn/problems/longest-well-performing-interval/description/
提示:
1 <= hours.length <= 10^4
0 <= hours[i] <= 16
解法1
class Solution {
public int longestWPI(int[] hours) {
int n = hours.length, d = 0, ans = 0;
Map<Integer, Integer> m = new HashMap<>();
m.put(0, 0);
for (int i = 0; i < n; ++i) {
if (hours[i] > 8) d++;
else d--;
if (d > 0) ans = Math.max(ans, i + 1);
else if (m.containsKey(d - 1)) {
ans = Math.max(ans, i - m.get(d - 1) + 1);
}
if (!m.containsKey(d)) m.put(d, i + 1);
}
return ans;
}
}
解法2——利用单调栈
class Solution {
public int longestWPI(int[] hours) {
int n = hours.length;
int[] s = new int[n + 1];
ArrayDeque<Integer> stk = new ArrayDeque<>();
stk.push(0);
// 单调递减的单调栈
for (int i = 1; i <= n; ++i) {
s[i] = s[i - 1] + (hours[i - 1] > 8? 1: -1);
if (s[i] < s[stk.peek()]) stk.push(i);
}
int ans = 0;
for (int i = n; i > 0; --i) {
while (!stk.isEmpty() && s[i] > s[stk.peek()]) {
ans = Math.max(ans, i - stk.pop());
}
}
return ans;
}
}