[AGC058D]Yet Another ABC String

StaroForgin

已于 2023-07-11 10:57:23 修改

阅读量914

点赞数 2

分类专栏： # 组合 ------容斥------ # 母函数文章标签：算法动态规划

于 2022-08-17 14:59:09 首次发布

本文链接：https://blog.csdn.net/Tan_tan_tann/article/details/126383405

版权

组合同时被 3 个专栏收录

42 篇文章 0 订阅

订阅专栏

------容斥------

16 篇文章 0 订阅

订阅专栏

母函数

6 篇文章 0 订阅

订阅专栏

这篇博客探讨了一种处理ABC字符串的优化动态规划（DP）方法。作者首先介绍了问题的背景，指出当有连续不合法部分时，可以将其转化为一段连续的ABC模式。接着，详细阐述了DP状态转移方程的建立，通过生成函数优化复杂度，最终给出一个时间复杂度为O(A+B+C)的解决方案。文章中还包含了源代码实现。

摘要由CSDN通过智能技术生成

Yet Another ABC String

题解

首先看到这个题，一个方向是去考虑容斥。
可以发现，如果有连续不合法部分的话，可以发现，他一定是一段连续的 $... A BC A BC A BC A BC ...$ 。
这是一个非常好的性质，因为如果我们之间容斥不合法位置的起点集合的话是非常麻烦的，这可以让我们得到一个非常好的转化。
每次强制原串中的一些部分呈现我们上面所说的这样的连续段，容斥我们这些连续段的集合。

考虑选择一个这样长度为 $n$ 的段，它所贡献容斥系数。
手推一下前几项可以发现，容斥系数在模 $3$ 为 $0$ 的位置上为 $- 1$ ，在模 $3$ 为 $1$ 的位置上为 $1$ ，模 $3$ 为 $2$ 的位置上为 $0$ 。
证明大概可以考虑通过递推式归纳证明，不是很会证明。（先留个坑）
好的，既然知道这东西了，我们也就很容易设计出来一个 $d p$ 转移。
由于有效的要么是模 $3$ 为 $0$ ，要么模 $3$ 为 $1$ ，相当于要么让所有数都减去同样的个数，要么在这基础上挑一个数多减一次。
定义 $dp_{i,j}$ 表示在构造一个长度为 $i$ 的串，让每个数减去 $\frac{i-j}{3}$ ，并且，进行 $j$ 次挑一个数多减 $1$ 的操作，得到的总系数和。
可以得到其转移式：
$dp_{i,j}=dp_{i-1,k-1}+\sum_{k=1}^{\min(i,3j+1)}[k\equiv 1\pmod{3}]dp_{i-k,j-\frac{k-1}{3}}-3[k\equiv 0\pmod{3}]dp_{i-k,j-\frac{k}{3}}$ 其中 $dp_{i-1,k-1}$ 表示这个位置没被选择容斥， $3$ 的系数是由于模 $3$ 为 $0$ 的情况有 $A BC, BC A, C A B$ 三种结构，模 $3$ 为 $1$ 的情况结构是固定的，没必要乘 $3$ 。
答案显然可以通过 $d p$ 值算出来，有
$Ans=\sum\binom{n-3i}{A-i,B-i,C-i}dp_{n,n-i}$ 。
再对上面的式子根据模 $3$ 余数加个前缀和优化，我们就得到一个简单的 $O\left(n^2\right)$ 的 $d p$ 做法了。

接下来让我们来看看怎么优化上面的这个 $d p$ ，不妨考虑生成函数。
定义二元生成函数 $G (x, y)$ ，其中 $x^iy^j]G$ 代表 $dp_{i,j}$ 的值。
通过 $G$ 的递推式可以得到，
$F=x+\sum_{i=1}^{\infty}x^{3i+1}y^i-3x^{3i}y^i=\frac{x-3x^3y}{1-x^3y}\\ G=\sum_{i=0}^{\infty} F^i=\frac{1}{1-F(x,y)}=\frac{1-x^3y}{1-x+2x^3y}\\$ 同样，考虑现在答案怎么计算。
$Ans=\sum_{i=0}^{\min(A,B,C)}\binom{n-3i}{A-i,B-i,C-i}[x^ny^i]G$ 令 $G'=\frac{1}{1-x+2x^3y}=\sum (x-2x^3y)^i$ ，显然有 $x^ny^m]G=[x^ny^m]G'-[x^{n-3}y^{m-1}]G'$ >
现在我们就只需要考虑 $x^ny^m]G'$ 怎么算了，这个好搞，可以解方程，看两边分别贡献了多少。
有 $a-3b=n,b=m\Rightarrow a=n-3m$ ，所以根据二项式定理可得，
$[x^ny^m]G'=\binom{n-2m}{m}(-2)^m$ 带到上面的式子里面去就可以快速算出答案了。

时间复杂度 $O\left(A+B+C\right)$ 。

源码

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#pragma GCC optimize("Ofast")
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int,int> pii;
#define MAXN 3000005
#define pb push_back
#define mkpr make_pair
#define fir first
#define sec second
const LL INF=0x3f3f3f3f3f3f3f3f;
const int mo=998244353;
template<typename _T>
void read(_T &x){
    _T f=1;x=0;char s=getchar();
    while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
    while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
    x*=f;
}
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
int add(int x,int y,int p){return x+y<p?x+y:x+y-p;}
void Add(int &x,int y,int p){x=add(x,y,p);}
int qkpow(int a,int s,int p){int t=1;while(s){if(s&1)t=1ll*a*t%p;a=1ll*a*a%p;s>>=1;}return t;}
int nA,nB,nC,n,fac[MAXN],inv[MAXN],ff[MAXN],pw2[MAXN],ans;
void init(){
    fac[0]=fac[1]=inv[0]=inv[1]=ff[1]=pw2[0]=1;
    for(int i=2;i<=n;i++)
        fac[i]=1ll*i*fac[i-1]%mo,
        ff[i]=1ll*(mo-mo/i)*ff[mo%i]%mo,
        inv[i]=1ll*ff[i]*inv[i-1]%mo;
    for(int i=1;i<=n;i++)pw2[i]=mo-add(pw2[i-1],pw2[i-1],mo);
}
int C(int x,int y){
    if(x<0||y<0||x<y)return 0;
    return 1ll*fac[x]*inv[y]%mo*inv[x-y]%mo;
}
int work(int x,int y){
    if(x<0||y<0||x<y+y)return 0;
    return 1ll*pw2[y]*C(x-y-y,y)%mo;
}
signed main(){
    read(nA);read(nB);read(nC);n=nA+nB+nC;init();
    for(int i=0;i<=min(nA,min(nB,nC));i++){
        int tmp=add(work(n,i),mo-work(n-3,i-1),mo);
        Add(ans,1ll*C(n-i*3,nA-i)*C(n-i*2-nA,nB-i)%mo*tmp%mo,mo);
    }
    printf("%d\n",ans);
    return 0;
}