Hot100-Day10-T438T438找到字符串中所有字母异位词

Day10[26/3/10]T438.找到字符串中所有字母异位词

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

示例 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" 的异位词。

提示:

• 1 <= s.length, p.length <= 3 * 104
• s 和 p 仅包含小写字母

解题思路

异位词的相同点在哪？排序后就完全相同了，或者用哈希表来看，就是每个字母出现的次数相同。

所以你可以从头到尾截取长度相等的字串，然后排序或者哈希之后，比较是否相等。

Code

#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_map>

using namespace std;

class Solution
{
public:
    vector<int> findAnagrams(string s, string p)
    {
        vector<int> result;

        if (s.size() < p.size())
            return result;

        string temp(p.size(), '\0');

        // sort(p.begin(), p.end());
        unordered_map<char, int> p_map;
        for (char c : p)
            p_map[c]++;

        for (int i = 0; i <= s.size() - p.size(); i++)
        {
            // 先拷贝
            temp = s.substr(i, p.size());
            // 排序后对比
            // sort(temp.begin(), temp.end());
            // if (temp == p)
            // result.push_back(i);
            auto p_map_temp = p_map;
            bool flag = true;
            for (char c : temp)
            {
                if (--p_map_temp[c] < 0)
                {
                    flag = false;
                    break;
                }
            }
            if (flag)
            {
                result.push_back(i);
            }
        }
        return result;
    }
};

auto main() -> int
{
    string s = "cbaebabacd", p = "abc";

    Solution sol;

    cout << "[";
    for (auto i : sol.findAnagrams(s, p))
    {
        cout << i << ",";
    }
    cout << "]" << endl;
}