2019.1.21【NOIP提高组】模拟B组 JZOJ 4209 已经没有什么好害怕的了

$Describe$

给定一个长度为$n$的括号序列，若一个子串是合法的，当且仅当所有括号都能被匹配，设每个位置经过的合法子串为$an{s}_i$，求${\sum }_{i=1}^n(an{s}_i\times imod10^9+7)$

---

$Solution$

用栈维护所有合法的位置，用差分数组求出经过的个数，需要注意的是，相邻的合法序列也为一个大的合法序列，所以要做两次前缀和，但这样会多算，要再用一个数组维护减去

---

$Code$

#include<cstring>
#include<cctype>
#include<cstdio>
#include<stack>
using namespace std;
long long ans[1000001],answer;
char str[1000001];
int t,n,l[1000002],r[1000002],len,p,q,pre[1000002],nxt[1000002];
inline void write(long long x){if(x>9)write(x/10);putchar(x%10+48);return;}
signed main()
{
	stack<int>s;
	scanf("%d",&t);
	while(t--)
	{
		while(s.size()) s.pop();answer=0;
		memset(l,0,sizeof(l));memset(r,0,sizeof(r));memset(ans,0,sizeof(ans));
		memset(pre,0,sizeof(pre));memset(nxt,0,sizeof(nxt));
		scanf("%s",str+1);n=strlen(str+1);
		for(register int i=1;i<=n;i++)
		{
			if(str[i]=='(') s.push(i);
			else if(s.size()) nxt[s.top()]=i+1,pre[i+1]=s.top(),s.pop();//找出所有左右括号的另一半所在的位置
		}
		for(register int i=n+1;i>0;i--) r[pre[i]]+=++r[i];//维护相邻的
		for(register int i=1;i<=n;i++) l[nxt[i]]+=--l[i];//因为是查分，所以范围之外的要减去
		for(register int i=1;i<=n;i++) {ans[i]=ans[~-i]+l[i]+r[i];answer+=(ans[i]*i)%1000000007;}//前缀和
		write(answer);putchar(10);//输出
	}
}