soj 131 找题
soj 131 找题
给出两个长度为n,都含k个1的字符串A,B。现在令\(a_1,a_2,\dots,a_k\)是A中1的下标,\(b_1,b_2,\dots,b_k\)是B中1的下表,然后将a,b等概率随机排列,接下来按1到k的顺序交换\(A_{a_i}\)与\(A_{b_i}\)。令P为交换之后A与B相同的概率,求\(P*(k!)^2\)对\(998244353\)取模的结果。n<=10000。
本来以为只要随机排列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;
}