给你一个二进制字符串 s
和一个整数 k
。
如果所有长度为 k
的二进制字符串都是 s
的子串,请返回 True ,否则请返回 False 。
输入: s = "00110110", k = 2 输出: true 解释: 长度为 2 的二进制串包括 "00","01","10" 和 "11"。它们分别是 s 中下标为 0,1,3,2 开始的长度为 2 的子串。
输入: s = "00110", k = 2 输出: true
输入: s = "0110", k = 1 输出: true 解释: 长度为 1 的二进制串包括 "0" 和 "1",显然它们都是 s 的子串。
输入: s = "0110", k = 2 输出: false 解释: 长度为 2 的二进制串 "00" 没有出现在 s 中。
输入: s = "0000000001011100", k = 4 输出: false
1 <= s.length <= 5 * 10^5
s
中只含 0 和 1 。1 <= k <= 20
use std::collections::HashSet;
impl Solution {
pub fn has_all_codes(s: String, k: i32) -> bool {
if s.len() < k as usize {
return false;
}
let mut x = i32::from_str_radix(&s[..k as usize], 2).unwrap();
let y = 1 << k;
let mut set = vec![x].into_iter().collect::<HashSet<_>>();
for c in s.bytes().skip(k as usize) {
x = (x << 1) % y + (c - b'0') as i32;
set.insert(x);
}
set.len() == y as usize
}
}