leetcode 696. 计数二进制子串 思维题

696. 计数二进制子串
     给定一个字符串 s，计算具有相同数量0和1的非空(连续)子字符串的数量，并且这些子字符串中的所有0和所有1都是组合在一起的。

重复出现的子串要计算它们出现的次数。

示例 1 :

输入: “00110011”
输出: 6
解释: 有6个子串具有相同数量的连续1和0：“0011”，“01”，“1100”，“10”，“0011” 和 “01”。

请注意，一些重复出现的子串要计算它们出现的次数。

另外，“00110011”不是有效的子串，因为所有的0（和1）没有组合在一起。

示例 2 :

输入: “10101”
输出: 4
解释: 有4个子串：“10”，“01”，“10”，“01”，它们具有相同数量的连续1和0。

注意：

s.length 在1到50,000之间。
s 只包含“0”或“1”字符。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-binary-substrings
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

• 考虑存一个数组a[maxn]，其中每个元素a[i]表示第i个连续串的个数
• 于是a[maxn]数组相邻两个元素的最小值的和就是答案
• 即 $ans={\sum }_{i=2}^{maxn}(min(a[i],a[i-1]))$
• 最后代码可以把a[maxn]数组优化掉

#define debug
#ifdef debug
#include 
#endif

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 

#define MAXN ((int)1e5+7)
#define ll long long int
#define INF (0x7f7f7f7f)
#define fori(lef, rig) for(int i=lef; i<=rig; i++)
#define forj(lef, rig) for(int j=lef; j<=rig; j++)
#define fork(lef, rig) for(int k=lef; k<=rig; k++)
#define QAQ (0)

using namespace std;

#define show(x...) \
    do { \
       cout << "\033[31;1m " << #x << " -> "; \
       err(x); \
    } while (0)

void err() { cout << "\033[39;0m" << endl; }
template<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }

namespace FastIO{

    char print_f[105];
    void read() {}
    void print() { putchar('\n'); }

    template <typename T, typename... T2>
       inline void read(T &x, T2 &... oth) {
           x = 0;
           char ch = getchar();
           ll f = 1;
           while (!isdigit(ch)) {
               if (ch == '-') f *= -1; 
               ch = getchar();
           }
           while (isdigit(ch)) {
               x = x * 10 + ch - 48;
               ch = getchar();
           }
           x *= f;
           read(oth...);
       }
    template <typename T, typename... T2>
       inline void print(T x, T2... oth) {
           ll p3=-1;
           if(x<0) putchar('-'), x=-x;
           do{
                print_f[++p3] = x%10 + 48;
           } while(x/=10);
           while(p3>=0) putchar(print_f[p3--]);
           putchar(' ');
           print(oth...);
       }
} // namespace FastIO
using FastIO::print;
using FastIO::read;

int n, m, Q, K;

int a[MAXN], sz;
class Solution {
public:
    int countBinarySubstrings(string s) {
        int n = s.length();
        if(!n) return 0;
        sz = 0;
        int cnt = 0, lst = s[0];
        for(int i=0; lst; ) { //统计a[]数组
            if(i == n || s[i] != lst) {
                a[++sz] = cnt;
                cnt = 0;
                lst = s[i];
            } else 
                cnt ++, i ++;
        }
        int sum = 0;
        for(int i=2; i<=sz; i++) //对于a[]里的每个相邻对的最小值累加就是答案
            sum += min(a[i], a[i-1]);
        return sum;
    }
};

#ifdef debug
signed main() {
    Solution s;
    cout << s.countBinarySubstrings("001100") << endl;





   return 0;
}
#endif