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。
题意:一个字符串,找到合法的串,将合法串的每一个位置都+1,最后*i(结果不用取模)
解析:首先,我们要对合法的子串有一个很好的理解()()这是3个合法的子串,设第一个括号为A,第二个为B,分别是:A,B,AB。也就是说,合法的串相连能构成一个新的合法串。所以说,要找到所有的串,利用差分的思想来做
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include <string>
#include <cstring>
using namespace std;
const int N=1e7+10;
char s[N];
long long l[N],r[N],jl[N],num1[N],n,f[N],num2[N];
int main()
{
int t;
cin>>t;
// string s;
for (int i=1;i<=t;i++)
{
scanf("%s",s+1),n=strlen(s+1);
long long t=0;
long long ans=0;//细节
for (int j=1;j<=1000001;j++) jl[j]=l[j]=r[j]=num1[j]=num2[j]=0;
for (int j=1;j<=n;j++)
{
if (s[j]=='(') jl[++t]=j;//记录左括号的位置
else if (t) l[j+1]=jl[t],r[jl[t--]]=j+1; // j+1为右括号的下一位,接下来有大用处
}
for (int j=n+1;j>0;j--)
num1[l[j]]+=num1[j]+1;//从右开始循环,找到右括号的下一位,加上右括号的下一位里的数,相当于加上了所有能与这个合法串组成新的合法串的个数,例如:有A B C 三个合法子串 A的情况:A加上B里面的数,也加上了能与B构成合法子串的数,能与B构成合法子串,也能与A构成合法子串。
for (int j=1;j<=n;j++) num2[r[j]]+=num2[j]-1;//和上面差不多
for (int j=1;j<=n;j++)
{
f[j]=f[j-1]+num1[j]+num2[j]; //num1[j]+num2[j]=f[j]-f[j-1]
ans+=(f[j]*j%1000000007);
}
cout<<ans<<endl;
}
}