-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.cpp
More file actions
64 lines (58 loc) · 2.02 KB
/
main.cpp
File metadata and controls
64 lines (58 loc) · 2.02 KB
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
// Source: https://leetcode.com/problems/count-binary-substrings
// Title: Count Binary Substrings
// Difficulty: Easy
// Author: Mu Yang <http://muyang.pro>
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Given a binary string `s`, return the number of non-empty substrings that have the same number of `0`'s and `1`'s, and all the `0`'s and all the `1`'s in these substrings are grouped consecutively.
//
// Substrings that occur multiple times are counted the number of times they occur.
//
// **Example 1:**
//
// ```
// Input: s = "00110011"
// Output: 6
// Explanation: There are 6 substrings that have equal number of consecutive 1's and 0's: "0011", "01", "1100", "10", "0011", and "01".
// Notice that some of these substrings repeat and are counted the number of times they occur.
// Also, "00110011" is not a valid substring because all the 0's (and 1's) are not grouped together.
// ```
//
// **Example 2:**
//
// ```
// Input: s = "10101"
// Output: 4
// Explanation: There are 4 substrings: "10", "01", "10", "01" that have equal number of consecutive 1's and 0's.
// ```
//
// **Constraints:**
//
// - `1 <= s.length <= 10^5`
// - `s[i]` is either `'0'` or `'1'`.
//
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#include <string>
using namespace std;
// Compute the size of 0-groups and 1-groups.
// For any two consecutive 0-group and 1-group (say with size a and b),
// we can form substrings using one 0&1, two 0&1, ..., min(a, b) 0&1;
// that is, min(a, b) substrings.
class Solution {
public:
int countBinarySubstrings(const string &s) {
char currCh = '\0';
int currSize = 0, prevSize = 0;
int ans = 0;
for (char ch : s) {
if (ch != currCh) {
ans += min(currSize, prevSize);
currCh = ch;
prevSize = currSize;
currSize = 0;
}
++currSize;
}
ans += min(currSize, prevSize);
return ans;
}
};