小郑的蓝桥平衡串（前缀和）

 问题描述

平衡串指的是一个字符串，其中包含两种不同字符，并且这两种字符的数量相等。

例如，abababababab 和 aababbaababb 都是平衡串，因为每种字符各有三个，而 abaababaab 和 aaaabaaaab 都不是平衡串，因为它们的字符数量不相等。

平衡串在密码学和计算机科学中具有重要应用，比如可以用于构造哈希函数或者解决一些数学问题。

小郑拿到一个只包含 LL、QQ 的字符串，他的任务就是找到最长平衡串，且满足平衡串的要求，即保证子串中 LL、QQ 的数量相等。

输入格式

输入一行字符串，保证字符串中只包含字符 LL、QQ。

输出格式

输出一个整数，为输入字符串中最长平衡串的长度。

样例输入

LQLL

样例输出

2

评测数据规模

对于所有评测数据，输入字符串的长度 len≤1000len≤1000。

运行限制

语言	最大运行时间	最大运行内存
C++	1s	256M
C	1s	256M
Java	2s	256M
Python3	3s	256M
PyPy3	3s	256M
Go	3s	256M
JavaScript	3s	256M

总通过次数: 3247  |  总提交次数: 3415  |  通过率: 95.1%

难度: 中等   标签: 字符串, 前缀和

代码

##小郑的蓝桥平衡串
##这里找出最长平衡字符串
from itertools import accumulate
str1=input()
a=[]
ans=0
##求前缀和
def get_presume(a):
    sum1=list(accumulate(a))
    return sum1
##有前缀和求区间长度
def get_sum(sum1,l,r):
##  l=0 从头开始
    if l==0:
        return sum1[r]
    else:
        return sum1[r]-sum1[l-1]
##这里只有两个 求区间长度 因此可以转化为前缀和的形式
##求前缀和又得需要l,r 故可以两边开始
for i in str1:
    if i=='L':
        a.append(1)
    else:
        a.append(-1)
##求得前缀和
b=get_presume(a)
n=len(a)
##根据前缀和数组 出入区间看是否等于0 表示平衡
for l in range(n):
    for r in range(l,n):
        if get_sum(b,l,r)==0:
##            r-l+1代表一个区间长度
            ans=max(ans,r-l+1)
print(ans)