CodeForces 873B Balanced Substring（前缀和+思维）

最新推荐文章于 2022-07-07 15:04:47 发布

原创最新推荐文章于 2022-07-07 15:04:47 发布 · 629 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

codeforces 同时被 2 个专栏收录

20 篇文章

订阅专栏

思想+思维

15 篇文章

订阅专栏

本文介绍了一种解决平衡子串问题的有效算法，通过使用前缀和的概念，将原本O(n^2)的时间复杂度降低到了O(n)，并提供了详细的实现步骤和源代码。

题目链接：点击打开链接

题意：给出的字符串由字符'0'和字符'1'组成，当一段区间[l,r]内的'0'和'1'的个数相同，就称为平衡子串，求最长的平衡子串。

思路：若利用前缀和记录0和1的个数（个数和，不是对应值的和），再去两层枚举，复杂度为O（n^2），会超时，所以要想出O（n）的算法，还是利用前缀和（值的和），'0'对应的值为-1，'1'对应的值为1，那么当一段前缀的和与另一段子前缀的和相等，也就是说相减得到的区间和为0，那么这段区间-1和1出现的次数相同，即'0'、'1'的个数相同，为平衡串。实现过程就是从头到尾扫一遍，每次更新前缀和，若该前缀和已经出现过，当前位置与第一次出现的位置做差，利用这段差区间的长度去更新结果，扫的过程同时要记录每个前缀和最早出现的位置。

// CodeForces 873B Balanced Substring 运行/限制：15ms/1000ms
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <iostream>
using namespace std;
#define INF 0x3f3f3f3f
int Min[200005];//Min[sum]表示前缀和sum的最早出现位置
char s[100005];
int main(){
	int n,sum,len,ans;
	while (scanf("%d", &n) != EOF) {
		scanf("%s", s + 1);
		sum = n;//防止sum为负数，sum初始化为n
		ans = 0;
		len = strlen(s + 1);
		memset(Min, INF, sizeof(Min));
		Min[sum] = 0;//重要细节
		for (int i = 1; i <= len; i++) {
			if (s[i] == '0') sum--;
			else sum++;
			Min[sum] = min(Min[sum], i);
			ans = max(ans, i - Min[sum]);
		}
		printf("%d\n", ans);
	}
    return 0;
}