Cthugha【前缀和】

>Link

luogu T210845


>Description

给出一个只包含 R 和 B 的字符串,求这个字符串中 R 的数量和 B 的数量相等的最长子串长度

∣ S ∣ ≤ 2 ∗ 1 0 6 |S|\le 2*10^6 S2106


>解题思路

维护一个前缀和,表示前缀子串中 R 比 B 多多少个
符合条件的子串要求我们 R 比 B 多0个,所以我们记录一下每个前缀和数值下,最小的下标 l l l 和最大的下标 r r r,用最大的减去最小的,显然 [ l + 1 , r ] [l+1,r] [l+1,r] 子串是符合条件的

然后记得,前缀和为 0 的最小下标为 0!!不然就30分了 万恶的捆绑测试


>代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 8000010
#define LL long long
using namespace std;

int n, sum[N];
LL minn[N], maxn[N], ans;
string s;

int main()
{
    
    
	cin >> s;
	n = s.size(), s = " " + s;
	memset (minn, 0x7f, sizeof (minn));
	minn[n] = 0;
	for (int i = 1; i <= n; i++)
	{
    
    
		sum[i] = sum[i - 1];
		if (s[i] == 'R') sum[i]++;
		else sum[i]--;
		minn[sum[i] + n] = min (minn[sum[i] + n], (LL)i);
		maxn[sum[i] + n] = max (maxn[sum[i] + n], (LL)i);
	}
	for (int i = 0; i <= 2 * n; i++)
	  if (maxn[i] > minn[i]) ans = max (ans, maxn[i] - minn[i]);
	printf ("%lld", ans);
	return 0;
}

Guess you like

Origin blog.csdn.net/qq_43010386/article/details/121374185