Codeforces Round 848 (Div. 2) E 期望dp

题目链接

总共n对，相同的记为x对，不相同的计为y对

每次等概率的改变一个值，也就是x/(x+y)的概率一个相同的变成不相同的

y/(x+y)的概率让一个不相同的变成相同的

然后我们要求的是不相同的从y对到0对的期望次数

记dp[i]为i+1个不相同的变成i个不相同的 的期望次数

很显然dp方程是这样的

然后我们都求出来之后，答案就是dp数组前y个的前缀和

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6+5;
#define int long long
const int INF = 4e18+5;
#define F(i,l,r) for(int i=l;i<=r;i++)
#define R(i,l,r) for(int i=r;i>=l;i--)
#define vv vector
#define fi first
#define se second
#define pii pair<int,int>
typedef long long ll;
const int mod = 998244353; 
//const int mod = 1e9+7; 
const int M = 1e7+5;
int a[N],b[N],c,d,vis[N],pri[N],ans1[N],ans2[N],dp[N],p[N][4]; 
int dx[4]={1,0,-1,0};
int dy[4]={0,1,0,-1};
int n,m,k,ans,x,y,n0,n2,m0,m2,op,op1,op2;
string s;
map<pii,int>mp;
map<int,int>mmap;
//char mp[1005][1005];
//vector<int>v1[N],v2[N];
int lowbit(int x) {
	return x&-x;
}
char xx[N];
bool temp;
int kuai(int x,int y){
	int sum=1;
	x%=mod;
	while(y){
		if(y%2){
			y--;
			sum*=x;
			sum%=mod;
		  }
	  x*=x;
	  y/=2;
	  x%=mod;
	}
	return sum;
}
struct node{
	int shu;
	int id;
}jie[N];
bool cmp(node x,node y){
	return x.shu<y.shu;
}
void solve()
{
	cin>>n;
	string s1,s2;
	cin>>s1>>s2;
	s1=" "+s1; s2=" "+s2;
	int ji=0,ou=0;
	F(i,1,n) if(s1[i]==s2[i]) ou++; else ji++;
	dp[n]=0; 
	R(i,0,n-1) dp[i]=((n-i-1)*dp[i+1]+n)%mod*kuai(i+1,mod-2)%mod;
	ans=0;
	F(i,0,ji-1) ans+=dp[i],ans%=mod;
	cout<<ans<<endl;
}
signed main() 
{
	ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0); // cin.tie(nullptr);
    int T = 1;                                                                                                                                                                                                                                                                    
    cin >> T;
    for (int i = 1; i <= T; i++)
    {
        solve();
    }
    return 0;
}
/*
6 4 3 1 2
1 1 4 5 1 4
*/