不难想到。考虑快排,发现是可以递归做的,然后就做完了。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#include<cstdio>
#include<algorithm>
#include<cstring>
typedef unsigned long long uLL;
using namespace std;
const int maxn=105,hP=10007,P=998244353;
int htab[maxn][hP+5],hnxt[100005],htot,f[1000005];
uLL vS[1000005],S_s,S_t,pw[maxn];
int& Find(int p,uLL S){
for(int j=htab[p][S%hP];j;j=hnxt[j]) if(vS[j]==S) return f[j];
vS[++htot]=S; hnxt[htot]=htab[p][S%hP]; htab[p][S%hP]=htot; f[htot]=-1;
return f[htot];
}
int n,m,a[3][maxn],top[3],b[maxn];
int dfs(int step,uLL S){
if(step<0) return 0;
int& res=Find(step,S);
if(res!=-1) return res;
res=(S==S_t);
for(int i=0;i<=2;i++)
if(top[i])
for(int j=0;j<=2;j++)
if(i!=j&&a[i][top[i]]<a[j][top[j]]){
int tmp=a[i][top[i]--];
a[j][++top[j]]=tmp;
res=(res+dfs(step-1,S+pw[tmp]*(j-i)))%P;
top[j]--; a[i][++top[i]]=tmp;
}

return res;
}
int main(){
scanf("%d%d",&n,&m);
pw[0]=1; for(int i=1;i<=n;i++) pw[i]=pw[i-1]*3;
for(int i=0;i<=n-1;i++){
scanf("%d",&b[i]); b[i]--;
S_s+=pw[i]*b[i]; a[b[i]][++top[b[i]]]=i;
}
a[0][0]=a[1][0]=a[2][0]=1e9;
for(int i=0;i<=2;i++) reverse(a[i]+1,a[i]+top[i]+1);
for(int i=0;i<=n-1;i++){
scanf("%d",&b[i]); b[i]--;
S_t+=pw[i]*b[i];
}
printf("%d\n",dfs(m,S_s));
return 0;
}