计数二进制子串——贪心算法

一、题目

给定一个字符串 s，计算具有相同数量0和1的非空(连续)子字符串的数量，并且这些子字符串中的所有0和所有1都是组合在一起的。重复出现的子串也要计算它们出现的次数。

二、解析

class SolutionS {
public:
	int countBinarySubstrings(string s) {
		vector<int> counts;//counts数组，统计相同字符串的个数
		int ptr = 0, n = s.size();
		while (ptr < n) {
			char c = s[ptr];
			int count = 0;
			while (ptr < n && s[ptr] == c) {
				++ptr;
				++count;
			}
			counts.push_back(count);
		}
		int ans = 0;
		for (int i = 1; i < counts.size(); ++i) {
			ans += min(counts[i], counts[i - 1]);
		}
		return ans;
	}
};

对于某一个位置 i，其实我们只关心 i−1位置的 counts 值是多少，所以可以用一个 last 变量来维护当前位置的前一个位置，这样可以省去一个 counts数组的空间。

class Solution {
public:
    int countBinarySubstrings(string s) {
        int ptr = 0, n = s.size(), last = 0, ans = 0;
        while (ptr < n) {
            char c = s[ptr];
            int count = 0;
            while (ptr < n && s[ptr] == c) {
                ++ptr;
                ++count;
            }
            ans += min(count, last);
            last = count;
        }
        return ans;
    }
};

参考：

https://leetcode-cn.com/problems/count-binary-substrings/solution/ji-shu-er-jin-zhi-zi-chuan-by-leetcode-solution/