438. 找到字符串中所有字母异位词（Leetcode）

给定一个字符串 s 和一个非空字符串 p，找到 s 中所有是 p 的字母异位词的子串，返回这些子串的起始索引。

字符串只包含小写英文字母，并且字符串 s 和 p 的长度都不超过 20100。

说明：

字母异位词指字母相同，但排列不同的字符串。
不考虑答案输出的顺序。
示例 1:

输入:
s: "cbaebabacd" p: "abc"

输出:
[0, 6]

解释:
起始索引等于 0 的子串是 "cba", 它是 "abc" 的字母异位词。
起始索引等于 6 的子串是 "bac", 它是 "abc" 的字母异位词。
 示例 2:

输入:
s: "abab" p: "ab"

输出:
[0, 1, 2]

解释:
起始索引等于 0 的子串是 "ab", 它是 "ab" 的字母异位词。
起始索引等于 1 的子串是 "ba", 它是 "ab" 的字母异位词。
起始索引等于 2 的子串是 "ab", 它是 "ab" 的字母异位词。

class Solution {
public:
	map<char, int>record(string s)
	{
		map<char, int>room;
		for (int i = 0; i < s.length(); ++i)
		{
			if (room.find(s[i]) != room.end())
			{
				room[s[i]]++;
			}
			else
				room[s[i]] = 1;
		}
		return room;
	}
	bool map_cmp(map<char, int>a, map<char, int>b)
	{
		for (map<char, int>::iterator it = a.begin(); it != a.end(); it++)
		{
			if (b.find(it->first) != b.end())
			{
				if (b[it->first] != it->second)
				{
					return 0;
				}
			}
			else
				return 0;
		}
		return 1;
	}
	vector<int> findAnagrams(string s, string p) {
		int N = p.length();

		vector<int>result;
		if (N > s.length())
			return result;

		map<char, int> room_s;
		map<char, int> room_p;
		room_s = record(s.substr(0, N));
		room_p = record(p);
		if(map_cmp(room_s, room_p))
			result.push_back(0);
		for (int i = 1; i < s.length() - N + 1; ++i)
		{
			room_s[s[i - 1]]--;
			if(room_s.find(s[i + N - 1]) != room_s.end())
				room_s[s[i + N - 1]]++;
			else
				room_s[s[i + N - 1]]=1;
			if (map_cmp(room_p, room_s))
				result.push_back(i);
		}
		return result;
	}
};