前缀和思想 宝石串（洛谷 P2697）

宝石串

题目描述

有一种宝石串，由绿宝石和红宝石串成，仅当绿宝石和红宝石数目相同的时候，宝石串才最为稳定，不易断裂。安安想知道从给定的宝石串中，可以截取一段最长的稳定的宝石串，有多少颗宝石组成。请你帮助他。

绿宝石用‘G’表示，红宝石用‘R’表示。

输入格式

一行由G和R组成的字符串

输出格式

最长的稳定的宝石串有多少颗宝石组成

这种基础题现在都写不出来，模板写再多也没有用，思维不跟上都白搭；

用‘G’代表+1，‘R’代表-1，然后计算前缀和，在记录每个前缀和的第一次出现位置；当这个前缀和值再出现时，减去第一次出现位置，更新答案；

注意前缀和可能为负；

代码：

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=1000010;
const int M=2000100;
const LL mod=2e9;
int n,sum[N],pos[M];
char s[N];
int main(){
	scanf("%s",s+1);
	for(int i=1;s[i]!='\0';i++){
		n++;
		sum[i]=sum[i-1];
		if(s[i]=='G') sum[i]++;
		else sum[i]--;
	}
	int ans=0;
	for(int i=1;i<=n;i++){
		if(sum[i]<0){
			int j=pos[sum[i]+N];
			if(j) ans=max(i-j,ans);
			else pos[sum[i]+N]=i;
		} 
		else if(sum[i]==0) ans=max(ans,i);
		else{
			int j=pos[sum[i]];
			if(j) ans=max(i-j,ans);
			else pos[sum[i]]=i;
		}
	}
	cout<<ans<<endl;
	return 0;
}