D. Flexible String Revisit(期望) Codeforces Round #848 (Div. 2)

题意：

给你两个二进制串a，b。你可以将a中的任意一位进行反转(0->1, 1->0)。问将a变成b操作次数的期望值是多少。

思路：

自己是看了幻想家协会会长: 讲解视频才会做的这道题，讲的很透彻。

首先我们先求出a和b其实有多少个位置不同用cnt来记录。如果cnt=0直接输出0就是了。
那么我们每一次操作无非就两种结果cnt+1， cnt-1。那么就此我们可以列出一个关于期望的式子f(x)，他的意思就是有x个位置不同变成相同的期望值。
$$f(x)=1+\frac{x}{n}f(x-1)+\frac{n-x}{n}f(n+1)$$
式子的讲解：首先你一定会操作一次，操作过后的效果无非就两种，第一你把相同的改成不同也就是$\frac{n-x}{n}f(n+1)$从相同中的去选，第二你把不同的改成相同的$\frac{x}{n}f(x-1)$从不同中的去选。

那么我们就此往下递推：
$$\begin{gathered} f(n)=f(n-1)+1<1> \\ . \\ f(n-1)=1+\frac{n-1}{n}f(n-2)+\frac{1}{n}f(n)<2> \end{gathered}$$
将1式带入2式，我们就可以得出：
$$\begin{gathered} f(n-1)=A\ast f(n-2)+B \\ . \\ . \\ f(k)=A_k\ast f(k-1)+B_k \end{gathered}$$
由此我们已知退出f(1),f(0)是等与0的那么$f(1)=B_1$，我们按照上式已知往下推我们就可以推出$B_1$
然后再有f(1)推出f(cnt)，求出答案；

代码：

#include<bits/stdc++.h>

#define IOS ios::sync_with_stdio(false);cin.tie(nullptr)
#define int long long
#define endl "\n"

using namespace std;

const int N = 2e6 + 10, mod = 998244353;

int ksm(int a, int b)
{
	int ans = 1;
	while(b) 
	{
		if(b & 1) ans =  ans*a %mod;
		a = a * a % mod;
		b >>= 1;
	}
	return ans;
}

int n, m, k, _;
char s1[N], s2[N];
int arr[N], brr[N];

void solve()
{
	cin >> n >> s1+1 >> s2+1;
	int cnt = 0;
	for(int i = 1; i <= n; i ++)
		if(s1[i] != s2[i]) cnt++;

	if(cnt == 0)
	{
		cout << 0 << endl;
		return;
	} 

	arr[n] = brr[n] = 1;

	for(int i = n-1; i >= 1; i --)
	{
		int t = ksm((n - (n-i) * arr[i+1]) % mod + mod, mod-2);
		arr[i] = i * t % mod;
		brr[i] = (n+ (n-i) * brr[i+1]) % mod * t % mod;
		//cout << t << " " << arr[i] << " " << brr[i] << endl;
	}
	int ans = brr[1];
	for(int i = 2; i <= cnt; i ++) ans = (ans * arr[i] + brr[i]) % mod;

	cout << ans << endl;

}

signed main()
{
	IOS;
	cin >> _;
	while(_--)
		solve();
	return 0;
}

补充

这里对于arr[i], brr[i]的推导代码进行讲述一下，自己最开始也很疑惑。
$$\begin{gathered} \because f(k)=A_k\ast f(k-1)+B_k; \\ . \\ f(n-i)=1+\frac{i}{n}\ast f(n-i-1)+\frac{n-i}{n}\ast f(n); \\ . \\ f(n-i)=1+\frac{i}{n}\ast f(n-i-1)+\frac{n-i}{n}\ast (A_i\ast f(i-1)+B_i); \\ . \\ f(n-i)=1+\frac{i}{n}\ast f(n-i-1)+\frac{n-i}{n}\ast A_i\ast f(i-1)+\frac{n-i}{n}\ast B_i; \\ . \\ (1-\frac{n-i}{n}\ast A_i)f(n-i)=\frac{i}{n}\ast f(n-i-1)+\frac{n-i}{n}\ast B_i+1; \\ . \\ \frac{n-(n-i)\ast A_i}{n}\ast f(n-i)=\frac{i}{n}\ast f(n-i-1)+\frac{(n-i)B_i+n}{n}; \\ . \\ f(n-i)=\frac{n}{n-(n-i)\ast A_i}\ast (\frac{i}{n}\ast f(n-i-1)+\frac{(n-i)B_i+n}{n}); \\ . \\ f(n-i)=\frac{1}{n-(n-i)\ast A_i}\ast (i\ast f(n-i-1)+(n-i)B_i+n); \end{gathered}$$
代码中的t = $\frac{1}{n-(n-i)\ast A_i}$就是这么来的。