[unknown OJ] 路径计数

一、题目

点此看题

二、解法

我人傻了，我是傻逼，考试时候 $C(x+y,x)$ 搞错成 $2^{x+y}$ 爆炸。

首先清楚关键问题是什么，先明确我们不可能把图构建出来，它只能帮助我们理解（然后由于是$csp$级别的考试后缀自动机也别去想了 ），有点复杂的就是对于一个子串，如果他的最大前缀后缀是$len-1$的话（也就是全部字符相等），那么他只能向自己的子串转移一次！其他情况都是$2$次，不难发现如果不存在相邻的相同字符，那么情况数是$2^n-1$的。

那么考虑减去这些算错的贡献，我们把所有相同字符段取出来，然后对于一个子串如果他的左边有$x$个字符，右边有$y$个字符，那么情况数是$C(x+y,x)$的，那么错算的方案数是：
$$\sum _{i=0}^{len-1}\sum _{j=0}^{len-1-i}C(x+y+i+j,x+i)$$可以结合杨辉三角进行化简，那么：
$$\sum _{i=0}^{len-1}C(x+y+len,x+i+1)-C(x+y+i,x+i+1)$$现在要考虑正确的方案数，如果取出来的这个串是作为一个整段出现的，那么贡献一定是：
$$len\times C(x+y,x)$$还有一些情况没有被考虑到，比如先选出$s[l-1,r]$再选出$s[l,l+i]$的，这一部分是没有被上面的整段计算进去的，同理$s[r-i,r]$也需要这样考虑，那么这些贡献是：
$$\sum _{i=1}^{len-1}(len-i)(C(x+y+i-1,x-1)+C(x+y+i-1,y-1))$$总之本题就是要算这些错算的段的贡献，时间复杂度$O(n)$

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int M = 500005;
const int MOD = 998244353;
#define int long long
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,ans,fac[M],inv[M];char a[M];
int C(int n,int m)
{
	return fac[n]*inv[m]%MOD*inv[n-m]%MOD;
}
void init(int n)
{
	fac[0]=inv[1]=inv[0]=1;
	for(int i=1;i<=n;i++) fac[i]=fac[i-1]*i%MOD;
	for(int i=2;i<=n;i++) inv[i]=inv[MOD%i]*(MOD-MOD/i)%MOD;
	for(int i=2;i<=n;i++) inv[i]=inv[i-1]*inv[i]%MOD;
}
int qkpow(int a,int b)
{
	int r=1;
	while(b>0)
	{
		if(b&1) r=r*a%MOD;
		a=a*a%MOD;
		b>>=1;
	}
	return r;
}
signed main()
{
	init(500000);
	scanf("%s",a+1),n=strlen(a+1);
	ans=qkpow(2,n)-1;
	for(int l=1,r=0,len=0;l<=n;l=r)
	{
		r=l;len=0;
		for(;a[l]==a[r];r++,len++);
		int x=l-1,y=n-r+1;
		for(int i=0;i<len;i++)
			ans=(ans-C(x+y+len,x+i+1)+C(x+y+i,x+i+1))%MOD;
		ans=(ans+len*C(x+y,x))%MOD;
		for(int i=1;i<len;i++)
			ans=(ans+(len-i)*(C(x+y+i-1,x-1)+C(x+y+i-1,y-1)))%MOD;
	}
	printf("%lld\n",(ans+MOD)%MOD);
}//abba