soj 131 找题

soj 131 找题

本来以为只要随机排列b就行了，但是由于是顺序交换，所以要随机排列必须a和b都排列一下。那怎么办呢？我们来举个栗子：

上面是a串，下面是b串。我们发现，蓝框内的位置，A数组都能交换两次。红框内的位置只能交换一次。

设i表示还有i个位置可以交换两次，j表示还有j个位置可以交换一次。\(f[i][j]\)表示剩下i和j的方案数，那么就有状态转移方程：\(f[i][j]=f[i-1][j]+f[i][j-1]\)。（我也想不清楚）。

https://blog.csdn.net/WerKeyTom_FTD/article/details/78209400

#include <cstdio>

#include <cstring>

using namespace std;

const int mod=998244353;

const int maxn=10005;

int l, x, y;

char s1[maxn], s2[maxn];

int fmi(int a, int x){

	long long base=a, ans=1;

	for (; x; base=base*base%mod, x>>=1)

		if (x&1) ans=ans*base%mod;

	return ans;

}

int jc[maxn], revjc[maxn];

void init(){

	jc[0]=revjc[0]=1;

	for (int i=1; i<maxn; ++i){

		jc[i]=1ll*jc[i-1]*i%mod;

		revjc[i]=fmi(jc[i], mod-2);

	}

}

int f[maxn][maxn/2], ans;

inline void up(int &x, int y){ x=(x+y)%mod; }

int c(int m, int n){ return 1ll*jc[m]*revjc[m-n]%mod*revjc[n]%mod; }

int main(){

	init(); scanf("%s%s", s1, s2); l=strlen(s1);

	for (int i=0; i<l; ++i)

		if (s1[i]==49&&s2[i]==49) ++x;

		else if (s1[i]==49||s2[i]==49) ++y;

	y/=2;

	f[x][y]=1;

	for (int i=x; i>=0; --i)

		for (int j=y; j>=0; --j){

			if (i) up(f[i-1][j], 1LL*i*j%mod*f[i][j]%mod);

			if (j) up(f[i][j-1], 1LL*j*j%mod*f[i][j]%mod);

		}

	for (int i=0; i<=x; ++i)

		up(ans, 1LL*jc[i]*jc[i]%mod*f[i][0]%mod*c(x+y, i)%mod);

	printf("%d\n", ans);

	return 0;

}