CF1264D1/2 Beautiful Bracket Sequence (easy/hard version)
这篇题解相对于其它题解对小白要友好一些。
模拟赛题，赛时 sb 了，\(n^2\) 都不会。
思路：
考虑什么情况下深度最大，容易发现 (((...))) 是肯定不劣的。
那么考虑枚举中心点的位置，设左边有 \(a\) 个左括号和 \(x\) 个问号，右边有 \(b\) 个右括号和 \(y\) 个问号，然后我们来枚举深度 \(i\)，求 \(i\) 的贡献次数。
要使得深度为 \(i\)，则要左边新添 \(i-a\) 个左括号，右边新添 \(i-b\) 个右括号，直接组合数计算贡献：
\[\sum_{i=\min(a,b)}^{\min(a+x,b+y)} \binom{x}{i-a} \binom{y}{i-b} i
\]
这样直接算是 \(O(N^2)\) 的，可以通过弱化版。
int main(){
init();
scanf("%s",s+1);
n=strlen(s+1);
For(i,1,n){
if(s[i]==')')
b++;
if(s[i]=='?')
y++;
}
For(i,1,n){
if(s[i]=='(')
a++;
if(s[i]=='?')
x++,y--;
if(s[i]==')')
b--;
l=min(a,b),r=min(a+x,b+y);
For(j,l,r)
ans=Add(ans,C(x,j-a)*C(y,j-b)%mod*j%mod);
}
write(ans);
return 0;
}
考虑拆式子为：
\[\Big(\sum_{i=0}^{n} \binom{x}{i-a} \binom{y}{i-b} (i-b) \Big)+ \Big(\sum_{i=0}^{n} \binom{x}{i-a} \binom{y}{i-b} b \Big)
\]
先看右边的式子，可以用范德蒙德卷积。
范德蒙德卷积基本形式：
\[\binom{n+m}{k} = \sum_{i=0}^k \binom{m}{i} \binom{n}{k-i} \]证明：
考虑组合意义，在 \(n+m\) 个物品中选 \(k\) 的方案数，是等价于在 \(n\) 个物品中选 \(i\) 个且在 \(m\) 个物品中选 \(k-i\) 个的总方案和的。
现在对于右边的式子：
\[b \sum_{i=0}^{n} \binom{x}{i-a} \binom{y}{i-b} = b \sum_{i=0}^{n} \binom{x}{x+a-i} \binom{y}{i-b}
\]
考虑组合意义后可化为：
\[b \binom{x+y}{x+a-b}
\]
现在看左边的式子：
\[\sum_{i=0}^{n} \binom{x}{i-a} \binom{y}{i-b} (i-b)
\]
注意到：
\[\begin{aligned} \binom{n}{m} m &= \frac{n!m}{m!(n-m)!} \\ &= \frac{n!}{(m-1)!(n-m)!} \\ &= \frac{(n-1)!}{(m-1)!(n-m)!} n \\ &= \binom{n-1}{m-1} n \end{aligned}
\]
那么左边式子可以化为：
\[\sum_{i=0}^{n} \binom{x}{i-a} \binom{y-1}{i-b-1} y = y \sum_{i=0}^{n} \binom{x}{x+a-i} \binom{y-1}{i-b-1}
\]
后面一串也可以考虑组合意义化简后得：
\[y \binom{x+y-1}{x+a-b-1}
\]
则对于 \(i\) 这个位置的总贡献为：
\[b \binom{x+y}{x+a-b} + y \binom{x+y-1}{x+a-b-1}
\]
现在时间复杂度优化为 \(O(N + \log P)\)。
需要提前预处理阶乘和阶乘逆元。
完整代码：
#include<bits/stdc++.h>
#define Add(x,y) (x+y>=mod)?(x+y-mod):(x+y)
#define lowbit(x) x&(-x)
#define pi pair<ll,ll>
#define pii pair<ll,pair<ll,ll>>
#define iip pair<pair<ll,ll>,ll>
#define ppii pair<pair<ll,ll>,pair<ll,ll>>
#define fi first
#define se second
#define full(l,r,x) for(auto it=l;it!=r;it++) (*it)=x
#define Full(a) memset(a,0,sizeof(a))
#define open(s1,s2) freopen(s1,"r",stdin),freopen(s2,"w",stdout);
#define For(i,l,r) for(int i=l;i<=r;i++)
#define _For(i,l,r) for(int i=r;i>=l;i--)
using namespace std;
typedef double db;
typedef unsigned long long ull;
typedef long long ll;
bool Begin;
const ll N=1e6+10,mod=998244353;
inline ll read(){
ll x=0,f=1;
char c=getchar();
while(c<'0'||c>'9'){
if(c=='-')
f=-1;
c=getchar();
}
while(c>='0'&&c<='9'){
x=(x<<1)+(x<<3)+(c^48);
c=getchar();
}
return x*f;
}
inline void write(ll x){
if(x<0){
putchar('-');
x=-x;
}
if(x>9)
write(x/10);
putchar(x%10+'0');
}
ll ans;
ll n,l,r,a,b,x,y;
ll f[N],inv[N];
char s[N];
ll qpow(ll a,ll b){
ll ans=1;
while(b){
if(b&1ll)
ans=(ans*a)%mod;
a=(a*a)%mod;
b>>=1ll;
}
return ans;
}
void init(){
f[0]=f[1]=1;
For(i,2,n)
f[i]=(f[i-1]*i)%mod;
inv[n]=qpow(f[n],mod-2);
_For(i,0,n-1)
inv[i]=inv[i+1]*(i+1)%mod;
}
ll C(ll n,ll m){
if(m>n||m<0||n<0)
return 0;
if(!m||m==n)
return 1;
return f[n]*inv[m]%mod*inv[n-m]%mod;
}
bool End;
int main(){
scanf("%s",s+1);
n=strlen(s+1);
init();
For(i,1,n){
if(s[i]==')')
b++;
if(s[i]=='?')
y++;
}
For(i,1,n){
if(s[i]=='(')
a++;
if(s[i]=='?')
x++,y--;
if(s[i]==')')
b--;
ans=Add(ans,(b*C(x+y,x+a-b)%mod+y*C(x+y-1,x+a-b-1)%mod)%mod);
}
write(ans);
return 0;
}