题目链接:展胜地的鲤鱼旗
设dp[i]为以 i 结尾的合法括号方案数。
那么对于当前的 i ,我们需要找到一个 i 结尾最早合法的括号,然后转移。最早结束的显然就是当前括号匹配的括号。
AC代码:
#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
//#define int long long
using namespace std;
const int N=1e6+10;
int n; long long dp[N],res; char str[N]; stack<int> s;
signed main(){
scanf("%s",str+1); n=strlen(str+1);
for(int i=1;i<=n;i++){
if(s.size()&&str[s.top()]=='('&&str[i]==')')
dp[i]=dp[s.top()-1]+1,s.pop();
else s.push(i);
res+=dp[i];
}
cout<<res;
return 0;
}