题目
剑指 Offer II 014. 字符串中的变位词 - 力扣(LeetCode)
思路
如果先计算出s1
的全排列,再看每个排列是否是s2
的子串,时间复杂度会非常高
如果两个字符串是变位词,不管是什么排列,其字母集一定相同,词频也一样
统计小写字母的词频可以用int[26]
,字母a对应下标0,依次类推
因为如果两个词互为变位词,其长度一定相等,因此我们检查s2中,长度为s1.length()
的子串即可
流程:
- 首先将s1的词频加到词频数组中
- 用两个指针指向子串的开头和结尾,每次把结尾往右移一位,将新加入的字母加到词频数组中,同时把开头指针往左移动一位,将其从磁盘数组中去除
- 检查这个子串是不是s1的变位词(即所有字母的词频是不是0)
代码
public boolean checkInclusion(String s1, String s2) {
// 如果s2的长度小于s1,说明s2一定不包含s1的变位词
if(s2.length() < s1.length()) {
return false;
}
int[] count = new int[26];
for (int i = 0;i<s1.length();i++) {
count[s1.charAt(i) - 'a']++;
count[s2.charAt(i) - 'a']--;
}
// 判断是否全为0
if (allZero(count)) {
return true;
}
for (int i = s1.length();i<s2.length();i++) {
// 加入s2.charAt(i)字符
count[s2.charAt(i) - 'a']--;
// 删除s2.charAt( i - s1.length()) 字符
count[s2.charAt( i - s1.length()) - 'a']++;
if (allZero(count)) {
return true;
}
}
return false;
}
private boolean allZero(int[] count) {
for (int each : count) {
if (each != 0) {
return false;
}
}
return true;
}