1. 程式人生 > 實用技巧 >演算法:字串的排列

演算法:字串的排列

描述

給定兩個字串s1和s2,寫一個函式來判斷 s2 是否包含 s1的排列。

換句話說,第一個字串的排列之一是第二個字串的子串。

示例1:

輸入: s1 = "ab" s2 = "eidbaooo"
輸出: True
解釋: s2 包含 s1 的排列之一 ("ba").

示例2:

輸入: s1= "ab" s2 = "eidboaoo"
輸出: False

注意:

輸入的字串只包含小寫字母
兩個字串的長度都在 [1, 10,000] 之間

連結:https://leetcode-cn.com/problems/permutation-in-string

思路

一個字串s的排列指s的每個字元出現都相等,可以用雜湊表來儲存s每個字元的次數,又由於只包含小寫字母,可考慮用一個長度26的陣列替代雜湊表。
然後可以用一個s長度的區間依次遍歷,檢查該區間是否每個字元頻次都與s相等。

程式碼:

class Solution {
public:
    bool checkInclusion(string s1, string s2) {
        if (s1.size() > s2.size()) return false;
        int count[26], tmp_count[26];
        memset(count, 0, sizeof count);
        memset(tmp_count, 0, sizeof tmp_count);
        for (const char& c : s1) {
            count[c-'a']++;
        }

        for (int i = 0, j = 0;j < s2.size();) {
            tmp_count[s2[j]-'a']++;
            if (j - i + 1 == s1.size()) {
                bool yes = true;
                for (int k = 0;k < 26;k++) {
                    if (tmp_count[k] != count[k]) {
                        yes = false;
                        break;
                    }
                }
                if (yes) return true;
                tmp_count[s2[i]-'a']--;
                ++i;
            }
            ++j;
        }
        return false;
    }
};

複雜度
時間複雜度:O(26N)
空間複雜度:O(1)