-
Notifications
You must be signed in to change notification settings - Fork 22
/
Copy path1461.check-if-a-string-contains-all-binary-codes-of-size-k.cpp
70 lines (65 loc) · 1.8 KB
/
1461.check-if-a-string-contains-all-binary-codes-of-size-k.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// Tag: Hash Table, String, Bit Manipulation, Rolling Hash, Hash Function
// Time: O(N)
// Space: O(N)
// Ref: -
// Note: -
// Given a binary string s and an integer k, return true if every binary code of length k is a substring of s. Otherwise, return false.
//
// Example 1:
//
// Input: s = "00110110", k = 2
// Output: true
// Explanation: The binary codes of length 2 are "00", "01", "10" and "11". They can be all found as substrings at indices 0, 1, 3 and 2 respectively.
//
// Example 2:
//
// Input: s = "0110", k = 1
// Output: true
// Explanation: The binary codes of length 1 are "0" and "1", it is clear that both exist as a substring.
//
// Example 3:
//
// Input: s = "0110", k = 2
// Output: false
// Explanation: The binary code "00" is of length 2 and does not exist in the array.
//
//
// Constraints:
//
// 1 <= s.length <= 5 * 105
// s[i] is either '0' or '1'.
// 1 <= k <= 20
//
//
class Solution {
public:
bool hasAllCodes(string s, int k) {
if (k > s.size()) {
return false;
}
unordered_set<string> visited;
int target = 1 << k;
for (int i = 0; i <= s.size() - k; i++) {
visited.insert(s.substr(i, k));
}
return visited.size() == target;
}
};
class Solution {
public:
bool hasAllCodes(string s, int k) {
int target = 1 << k;
unsigned int mask = target - 1;
unsigned int hashing = 0;
vector<bool> visited(target, false);
int find = 0;
for (int i = 0; i < s.size(); i++) {
hashing = ((hashing << 1) & mask) | (s[i] - '0');
if (i >= k - 1 && !visited[hashing]) {
visited[hashing] = true;
find += 1;
}
}
return find == target;
}
};