CF 990C. Bracket Sequences Concatenation Problem【栈/括号匹配】

【链接】:CF
【题意】:
给出n个字符串,保证只包含'('和')',求从中取2个字符串链接后形成正确的括号序列的方案数(每个串都可以重复使用)(像'()()'和'(())'这样的都是合法的,像')('和'('这样的是不合法的)
输入:
第一行一个整数n
第二行到第n+1行每行一个字符串

输出:
方案数
【分析】:
1.本来就正常的,只能和正常匹配的一起

2.本来就不正常匹配的,只能和不正常匹配的在一起

那么对于每一个串,我们处理出它还需要cnt1个'('和cnt2个')'

如果cnt1,cn2>0,不管在左边还是右边加串,都不可行

如果cnt1==0&&cnt2==0已经是正常匹配

如果cnt1>0,cnt2==0,该串的右边需要cnt1个')'

如果cnt1==0,cnt2>0,该串的左边需要cnt2个')'

那么只要把需要cnt个左边的和cnt个右边的放在一起就一定可以匹配上了

复杂度O(n)
【代码】:

#include<bits/stdc++.h>
#define PI acos(-1.0)
#define pb push_back
#define F first
#define S second
#define debug puts
#define setp cout << fixed << setprecision(3)
#define FAST_IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
typedef long long ll;
const int N=3e5+5;
const int MOD=1e9+7;
const ll INF=1e18+8;
ll cntl[N],cntr[N];
int main(void){
    FAST_IO;
    int n;
    cin >> n;
    ll ans=0;
    for(int i=1;i<=n;i++){
        string s;
        cin >>s;
        int l=0,r=0;
        for(int i=0;s[i];i++){
            if(s[i]=='(') l++;
            else{
                if(l>0) l--; 
                else    r++;
            }
        }
        //cout<<l<<' '<<r<<endl;
        if(l==0&&r==0)  ans++;
        else if(l!=0&&r!=0) continue;
        else if(l==0&&r!=0) cntl[r]++;
        else if(l>0&&r==0)  cntr[l]++;
    }
    ans*=ans;
    for(int i=1; i<=3e5; i++){
        if(cntl[i]>0 && cntr[i]>0)    
            ans+=cntl[i]*cntr[i];
    }
    cout << ans << endl;
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/Roni-i/p/9215513.html