【Leetcode每日一题】1221. 分割平衡字符串

题目

在一个 平衡字符串 中，‘L’ 和 ‘R’ 字符的数量是相同的。

给你一个平衡字符串 s，请你将它分割成尽可能多的平衡字符串。

注意：分割得到的每个字符串都必须是平衡字符串。

返回可以通过分割得到的平衡字符串的 最大数量 。

示例

示例1

输入：s = “RLRRLLRLRL”
输出：4
解释：s 可以分割为 “RL”、“RRLL”、“RL”、“RL” ，每个子字符串中都包含相同数量的 ‘L’ 和 ‘R’ 。

示例2

输入：s = “RLLLLRRRLR”
输出：3
解释：s 可以分割为 “RL”、“LLLRRR”、“LR” ，每个子字符串中都包含相同数量的 ‘L’ 和 ‘R’ 。

示例3

输入：s = “LLLLRRRR”
输出：1
解释：s 只能保持原样 “LLLLRRRR” 。

示例4

输入：s = “RLRRRLLRLL”
输出：2
解释：s 可以分割为 “RL”、“RRRLLRLL” ，每个子字符串中都包含相同数量的 ‘L’ 和 ‘R’ 。

关键思路

考虑字符串第 i 个到第 j 个字符组成的子串中 ‘L’ 和 ‘R’ 的数量是否相等，代码中使用 python 自带的 count() 函数对 ‘L’ 的数量进行统计，判断其是否等于当前子字符串长度的一半。

i 初始值为0，即从输入字符串的头开始考虑；若当前子字符串符合要求，就将 i 重新设置为 j+1, 即对字符串进行划分。

为了提高程序效率，我们每次循环中递增的字符串长度设置为 2。例如，对于‘RLRRLLRLRL’， 每轮检查的顺序依次为：
（1） s[0:2]：‘RL’
（2） s[2:4]： ‘RR’
（3） s[2:6]： ‘RRLL’
（4） s[6:8]： ‘RL’
（5） s[8:10]： ‘RL’

代码实现

class Solution(object):
    def balancedStringSplit(self, s):
        """
        :type s: str
        :rtype: int
        """

        i, n = 0, 0

        for j in range(2, len(s)+1, 2):
            if s.count('L', i, j) == len(s[i:j]) / 2 :
                i = j  # reset the index
                n = n+1  # add

        return n


if __name__ == "__main__":
    s = input()
    obj = Solution()
    result = obj.balancedStringSplit(s)
    print(result)

运行结果

4
3
1
2

链接

https://leetcode-cn.com/problems/split-a-string-in-balanced-strings