解法一
用前缀和的思想统计从第一位到其他当前位置的各个元音字母出现的次数。然后用5位二进制数对5个元音字母的奇偶状态进行编码,放进一个List,存储当前下标。最后针对各编码的下标数组(即不同的元音字母奇偶状态)计算其下标最大值和最小值之差,即可得到满足条件的子串的长度,再从中找出最大值即为答案。
import java.util.List;
import java.util.Map;
import java.util.HashMap;
import java.util.LinkedList;
class Solution {
public int findTheLongestSubstring(String s) {
// 防止第一位影响计算
s = 'b' + s;
// 统计到当前位数各个元音字母的数量
int[][] count = new int[s.length()][5];
// 将有相同编码的下标放在同一数组中
Map<Integer, List<Integer>> indexMap = new HashMap<>();
// 用于计算编码
int key = 0;
for (int i = 0; i < s.length(); ++i) {
if (i > 0) {
count[i] = count[i - 1].clone();
}
switch (s.charAt(i)) {
case 'a':
++count[i][0];
break;
case 'e':
++count[i][1];
break;
case 'i':
++count[i][2];
break;
case 'o':
++count[i][3];
break;
case 'u':
++count[i][4];
break;
}
// 根据各个元音字母个数奇偶情况编码
key = 0;
for (int j = 0; j < 5; ++j) {
if (count[i][j] % 2 == 1) {
key <<= 1;
++key;
} else {
key <<= 1;
}
}
if (indexMap.containsKey(key)) {
indexMap.get(key).add(i);
} else {
LinkedList<Integer> list = new LinkedList<>();
list.add(i);
indexMap.put(key, list);
}
}
// 统计根据相同编码的对应下标计算满足条件的最大子串长度
int maxLength = 0;
for (List<Integer> list : indexMap.values()) {
maxLength = Math.max(
maxLength,
list.get(list.size() - 1) - list.get(0)
);
}
return maxLength;
}
}
解法二
解法一是自己想完做出来的,但是一提交发现时间内存都比较糟糕,再去看官方题解:https://leetcode-cn.com/problems/find-the-longest-substring-containing-vowels-in-even-counts/solution/mei-ge-yuan-yin-bao-han-ou-shu-ci-de-zui-chang-z-2/
其实前缀和和状态压缩的思想在解法一中都已经想到了,但是编码实现的方式上可以进一步提升。通过异或操作可以直接利用上一次奇偶状态,而不同编码对应的下标只需要存储最小值,也无需用复杂的Map。
import java.util.Arrays;
class Solution {
public int findTheLongestSubstring(String s) {
// 防止第一位影响计算
s = 'b' + s;
// 满足编码的最小下标
int[] keyArr = new int[1 << 5];
Arrays.fill(keyArr, -1);
// 当前奇偶状态
int key = 0;
// 最大长度
int maxLength = 0;
for (int i = 0; i < s.length(); ++i) {
switch (s.charAt(i)) {
case 'a':
key ^= 1 << 4;
break;
case 'e':
key ^= 1 << 3;
break;
case 'i':
key ^= 1 << 2;
break;
case 'o':
key ^= 1 << 1;
break;
case 'u':
key ^= 1 << 0;
break;
}
if (keyArr[key] >= 0) {
maxLength = Math.max(maxLength, i - keyArr[key]);
} else {
keyArr[key] = i;
}
}
return maxLength;
}
}
题外
在第一遍做的时候,我一开始想用泛型数组来做不同状态的下标数组,但是由于Java泛型的实现机制,泛型数组无法使用,所以用了Map。关于这个点可以参看以下知乎讨论。