题目描述:
已知一个长度为N的字符串,只由0和1组成, 求一个最长的子串,要求该子串出0和1出现的次数相等。
要求算法时间复杂度尽可能的低。
比如: 1000010111000001,加粗的部分有4个0、4个1
思路:
(1) 最简单的想法就是遍历所有的子串,之后判断该子串是否满足条件
N^2子串,每个子串扫一遍判断0、1是否出现的次数相等,复杂度为O(N^3)
稍加思考就会发现, 如果一个长度为n的子串满足条件,加么这n个元素的和 加起来一定=(n/2)
这样在循环的过程中,增量加就可以了,不需要每个子串从头计算,复杂度降为O(N^2);
伪码:
输入 A[N]
int maxlen = 0, sum = 0, currlen = 0;
for(int i = 0; i < N; ++i)
{
sum = 0;
for(int j = i; j < N; ++j)
{
currlen = j - i + 1;
sum += int(A[j]);
if(currlen%2 == 0 && sum == currlen/2 && currlen > maxlen)
maxlen = currlen;
}
for(int i = 0; i < N; ++i)
{
sum = 0;
for(int j = i; j < N; ++j)
{
currlen = j - i + 1;
sum += int(A[j]);
if(currlen%2 == 0 && sum == currlen/2 && currlen > maxlen)
maxlen = currlen;
}