【五校联考1day1】已经没有什么好害怕的了

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<cstdio>
#include<cstring>
#define mod 1000000007
#define ll long long 
using namespace std;
const int N=1e6+77;
ll T,len,f[N],num1[N],num2[N],k[N],l[N],r[N],ans;
char s[N];
int main()
{
	scanf("%lld",&T);	
	while (T--)
	{
		scanf("%s",s+1),len=strlen(s+1);
		for(int i=1; i<=len; i++) if(s[i]=='(') k[++k[0]]=i;
		else if(k[0]) r[k[k[0]]]=i+1,l[i+1]=k[k[0]--];
		for(int i=len+1; i; i--) num1[l[i]]+=++num1[i];
		for(int i=1; i<=len; i++) num2[r[i]]+=--num2[i];
		for(int i=1; i<=len; i++) f[i]=f[i-1]+num1[i]+num2[i];
		for(int i=1; i<=len; i++) ans=ans+f[i]*i%mod;
		printf("%lld\n",ans);
		memset(f,0,sizeof(f)); memset(num1,0,sizeof(num1)); memset(num2,0,sizeof(num2)),
		memset(l,0,sizeof(l)); memset(r,0,sizeof(r)); memset(k,0,sizeof(k)),ans=0;
	}
}