jzoj4209.已经没有什么好害怕的了(B组——Day4)
题目
Description
小Y 最近开始学习算法姿势,但是因为小R 非常BB,给了她很多B6 题,所以她觉得自己已经没有什么前途了。于是小R 给了她一些稍微简单的题,让她觉得已经没有什么好害怕的了,其中一道是这样的:
给定一个长度为n 只包含左括号和右括号的序列,现在小R 想要知道经过每一个位置的合法子串有多少个。
空串是一个合法的串,如果A 和B 都是合法的串,那么(A) 和AB 都是合法的串。
Input
第一行输入一个正整数T 表示数据组数。接下来T 行每行一个字符串。
Output
对于每组数据,输出一个整数表示答案,令ansi 为经过第i 个位置的子串个数,那么你需要输出(注意是先求余再求和)
Sample Input
1
()()
Sample Output
20
样例解释:
ans 数组为{2,2,2,2},所以输出20。
Data Constraint
对于10% 的数据,n<=100
对于30% 的数据,n <= 1000
对于60% 的数据,n <= 5 <= 10^4
对于100% 的数据,n <= 10^6,1 <= T<= 10
解析
代码:
#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1000001;
long long f[N],sl[N],sr[N],tail[N],l[N],r[N];
char s[N];
int main(){
long long t;
scanf("%lld",&t);
while (t>0){
t--;
long long ans=0,len;
scanf("%s",s+1);
len=strlen(s+1);
for(int i=0;i<=N;++i)
sl[i]=sr[i]=tail[i]=l[i]=r[i]=f[i]=0;
for(int i=1;i<=len;i++)
if(s[i]=='(') tail[++tail[0]]=i;
else if(tail[0]) r[tail[tail[0]]]=i+1,l[i+1]=tail[tail[0]--];
for(int i=len+1;i;i--) {
sl[i]++;
sl[l[i]]+=sl[i];
}
for(int i=1;i<=len;i++) {
sr[i]--;
sr[r[i]]+=sr[i];
}
for(int i=1;i<=len;i++) {
f[i]=f[i-1]+sl[i]+sr[i];
ans=ans+f[i]*i%1000000007;
}
printf("%lld\n",ans);
}
}