Leetcode 5409：检查一个字符串是否包含所有长度为K的二进制子串

题目描述

给你一个二进制字符串 s 和一个整数 k 。

如果所有长度为 k 的二进制字符串都是 s 的子串，请返回 True ，否则请返回 False 。

 

示例 1：

输入：s = "00110110", k = 2
输出：true
解释：长度为 2 的二进制串包括 "00"，"01"，"10" 和 "11"。它们分别是 s 中下标为 0，1，3，2 开始的长度为 2 的子串。

示例 2：

输入：s = "00110", k = 2
输出：true

示例 3：

输入：s = "0110", k = 1
输出：true
解释：长度为 1 的二进制串包括 "0" 和 "1"，显然它们都是 s 的子串。

示例 4：

输入：s = "0110", k = 2
输出：false
解释：长度为 2 的二进制串 "00" 没有出现在 s 中。

示例 5：

输入：s = "0000000001011100", k = 4
输出：false

 

提示：

• 1 <= s.length <= 5 * 10^5
• s 中只含 0 和 1 。
• 1 <= k <= 20

 

 

解题思路

class Solution {
public:
    unordered_set st;
    bool dfs(int cur, int k, string &str){
        if(cur >= k){
            return st.find(str) != st.end();
        }
        str[cur] = '0';
        bool b = dfs(cur+1, k, str);
        if(!b) return false;
        str[cur] = '1';
        return dfs(cur+1, k, str);
    }
    bool hasAllCodes(string s, int k) {
        string str(k,'0');
        int len = s.length();
        if(k > len) return false;
        for(int i=k; i<=len; ++i){
            string tmp = s.substr(i-k,k);
            st.insert(tmp);
        }
        bool ans = dfs(0, k, str);
        return ans;
    }
};