做题链接👇👇👇
面试题 17.05. 字母与数字
题解:题目需要我们返回字母与数字个数相同的最长子数组,我们首先要考虑在子数组中如何找出数字与字母个数相同。
这里我们可以使用前缀和的思路来做,将字母转换成+1,数字转换成-1,如果当前子数组的前缀和为0那么字母与数字的个数也就相同。
如果前缀和出现多次,我们为了达成最长子数组这一特点,我们只需要记录不同前缀和第一次出现的位置。
这里使用哈希表来记录每一个前缀和第一次出现的下标,由于最后的答案需要返回一个数组我这边使用Arrays
工具类的复制数组方法copyOfRange
。最后注意一下边界的问题,因为copyOfRange区间是左闭右开的,所以记得要将当前索引加1作为右区间。但这里不能使用copyOf方法,因为copyOf是从第一个索引开始复制的
代码:
class Solution {
public String[] findLongestSubarray(String[] array) {
HashMap<Integer, Integer> map = new HashMap<>();
map.put(0, 0);
int start = 0, end = 0, n = array.length, count = 0;
for (int i = 0; i < array.length; i++) {
if (Character.isLetter(array[i].charAt(0))) {
count++;
} else {
count--;
}
//如果有这个前缀和就看看是否此子数组比之前子数组长度更长,如果没有就将此前缀和存入哈希表
if (map.containsKey(count)) {
Integer a = map.get(count);
if (i + 1 - a > end - start) {
start = a;
end = i + 1;
}
} else {
map.put(count, i + 1);
}
}
return Arrays.copyOfRange(array,start,end);
}
}