LeetCode hoot 100 -- 找到字符串中的所有字母异位词

本题取自LeetCode hoot 100 题号438 找到字符串中的所有字母异位词

一 题目概述

给定两个字符串 s 和 p，找到 s中所有 p的 异位词的子串，返回这些子串的起始索引。不考虑答案输出的顺序。

示例 1:

输入: s = "cbaebabacd", p = "abc"
输出: [0,6]
解释:
起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。

二 思路解析

遇到类似判断子串问题，很容易想到双重循环求子串，再判断是否满足条件。时间复杂度过高，在此我们可以利用 '滑动窗口' ，顾名思义，设置一个子串大小的窗口，并不断向前推进。

不妨以题目所给示例加以说明：

s = "cbaebabacd", p = "abc" 规定初始窗口在前三个字符cba 考虑到需要判断是否是异位词，就是要判断每个字母出现的次数是否和p相同，我们不妨设置两个数组，一个记录p中各个字母出现次数，一个记录s的子串中各个字母出现次数。

很显然cba 符合abc的异位词。向前推进至 bce 不符合异位词 abc。依次类推 下面是图示说明：

三 代码解析

#include <iostream>
#include <vector>
#include <map>
using namespace std;

int main()
{
	string s, p;
	cin >> s >> p;
	vector<int>res;//记录匹配位置
	map<char, int>pMap;//记录模式串中每个字符出现的次数
	map<char, int>sMap;//记录主串中每个字符出现的次数
	//统计模式串中每个字符出现的次数
	for (char c : p)
	{
		pMap[c]++;
	}

	for (int i = 0; i < p.size(); i++) sMap[s[i]]++;//初始化窗口，统计主串中前p.size()个字符出现的次数
	if (sMap == pMap) res.push_back(0);//如果初始窗口匹配，记录位置0

	//滑动窗口，统计主串中每个字符出现的次数 i记录起始索引 注意数组越界
	for (int i = 1; i <= s.size()-p.size(); i++)
	{
		//窗口向右滑动，更新sMap
		sMap[s[i-1]]--;
		if (sMap[s[i - 1]] == 0) sMap.erase(s[i - 1]);
		sMap[s[i + p.size() - 1]]++;
		if (sMap == pMap) res.push_back(i);
		
	}
	for (int i = 0; i < res.size(); i++)
	{
		cout << res[i] << " ";
	}
	return 0;
}

值得注意的是 我利用map记录字母（key）和其对应的出现次数（value），然而本题只出现了小写字母，我们可以利用数组替代map，数组大小为26 对应26个小写字母。数组效率一般高于map。

感谢阅读。