题目:https://www.luogu.org/problemnew/show/CF990C
题意: 输入n个字符串,每个字符串都只有 ( 和 ) 组成,求从中取2个字符串链接后形成正确的括号序列的方案数。
题解:
所有的字符串可以分为3类:
1: 自身完美匹配型(即左括号和右括号完美匹配)用ans来标记。
2:除去完全匹配的子串,剩下的都是左括号,用数组l来标记。
3:除去完全匹配的子串,剩下的都是右括号,用数组r来标记。
l[left]表示 (个数为left的 字符串个数;
r[right]表示 )个数为right的 字符串个数;
对于完美匹配型,方案数目为:ans*ans;
对于另外两种类型,方案数目为:l[x]*r[x] (即数目为x的仅含左括号字符串和仅含右括号字符串);
代码如下:
#include<iostream> //R
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
const int maxn=3e5+1;
char s[maxn];
long long int l[maxn],r[maxn];
using namespace std;
int main()
{
std::ios::sync_with_stdio(false);
int n;
cin>>n;
long long ans=0;
while(n--)
{
cin>>s;
int lefts=0,rights=0;
for(int i=0;i<strlen(s);i++){
if(s[i]=='(') lefts++;
else{
if(lefts>0) lefts--;
else rights++;
}
}
if(lefts==0&&rights==0) ans++;
else if(lefts!=0&&rights==0) r[lefts]++;
else if(lefts==0&&rights!=0) l[rights]++;
else if(lefts!=0&&rights!=0) continue;
}
ans*=ans;
for(int i=0;i<3e5+1;i++){
if(l[i]>0&&r[i]>0)
ans+=l[i]*r[i];
}
cout<<ans;
return 0;
}