【差分】JZOJ_4209 已经没有什么好害怕的了

题意

给出一个括号字符串。
空串是一个合法的串，如果$A$是一个合法的串，那么$(A)$也是一个合法的串。如果$B$也是一个合法的串，那么$AB$也是一个合法的串。
令$an{s}_i$为经过第$i$个位置的子串个数，求

思路

我们可以用差分然后求前缀和的方法做。
对于每个合法的串，我们可以把它两侧的合法的串的贡献加上去。减的时候也把它们减去。如果遇到不匹配的括号它会自动抵消掉。

代码

#include<cstdio>
#include<cstring>

int t;
long long ans;
int w[1000001], cl[1000001], cr[1000002], l[1000001], r[1000001];
long long pre[1000001];
char s[1000001];

long long input() {
	char c = getchar();
	int f = 1;
	long long result = 0;
	while (c < '0' || c > '9') {
		if (c == '-') f = -1;
		c = getchar();
	}
	while (c >= '0' && c <= '9') {
		result = result * 10 + (c - 48);
		c = getchar();
	}
	return result * f;
}

void write(long long x) {
	if (x > 9) write(x / 10);
	putchar(x % 10 + 48);
}

void print(long long x) {
	if (x < 0) printf("-"), x *= -1;
	write(x);
}

void println(long long x) {
	print(x);
	putchar(10);
}

void solve() {
	memset(cl, 0, sizeof(cl));
	memset(cr, 0, sizeof(cr));
	memset(l, 0, sizeof(l));
	memset(r, 0, sizeof(r));
	scanf("%s", s + 1);
	int n = strlen(s + 1), t = 0;
	ans = 0;
	for (register int i = 1; i <= n; i++)
		if (s[i] == '(') w[++t] = i;
		else if (t) {
			r[w[t]] = i + 1;
			l[i + 1] = w[t--];
		}//括号匹配
	for (register int i = n + 1; i >= 1; i--)
		cr[l[i]] += ++cr[i];
	for (register int i = 1; i <= n; i++)
		cl[r[i]] += --cl[i];
	pre[0] = 0;
	for (register int i = 1; i <= n; i++)
		pre[i] = pre[i - 1] + cl[i] + cr[i];
	for (register int i = 1; i <= n; i++)
		ans += i * pre[i] % 1000000007;
	println(ans);
}

int main() {
	t = input();
	for (; t; t--, solve());
}