洛谷P2697 宝石串

洛谷P2697 宝石串

#include <bits/stdc++.h> 
#define For(i,j,k) for(int i=j;i<=k;i++) 
using namespace std ; 

const int N = 1000011,inf = 1e9 ; 
int n,ans ; 
char s[N] ; 
int sum[N],t[N],l[N],tt[N],r[N]  ; 

inline int read() 
{
    int x = 0 , f = 1 ; 
    char ch = getchar() ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x * 10+ch-48 ; ch = getchar() ; } 
    return x * f ;  
}

int main() 
{
    scanf("%s",s+1) ; 
    n = strlen(s+1) ; 
    For(i,1,n) 
        sum[ i ]=sum[i-1]+((s[i]=='R')?-1:1) ;  
    For(i,-N,N) l[ i ]=N ; 
    l[ 0 ] = 0 ; 
    For(i,1,n) {
        if(l[ sum[i] ]==N) l[sum[i]] = i ; 
        r[sum[i]] = i ; 
    }
    ans = 0 ; 
    For(i,-n,n) 
        if( r[i]-l[i]>ans ) ans = r[i]-l[i] ; 
    printf("%d
",ans) ; 
    return 0 ; 
}