UOJ【UR #11】元旦老人与汉诺塔 题解

题目传送门

题目大意： 给出汉诺塔的起始状态和最终状态，问有多少种走法在 $m$ 步以内从起始状态到达最终状态。

题解

跑一发暴力可以发现，要将一根柱子上的 $7$ 个圆盘全部移到另一个柱子上，需要 $127$ 步，也就是说，$100$ 步以内最多移一个柱子的 $7$ 个圆盘，三根柱子就是 $21$ 个，而仔细想想发现，每次只能移动最上面的圆盘，也就是说，一个状态只有三个后续状态，所以总状态数大约就是 $3^{21}$ 种，但其中又有很多像把一个圆盘移走又移回来的操作，所以用一个记忆化搜索就能过。

代码如下（太丑了各位就别看了qwq……）：

#include <cstdio>
#include <cstring>
#include <string>
#include <map>
#include <iostream>
#include <algorithm>
using namespace std;
#define maxn 110
#define mod 998244353

int n,m,a[maxn];
string S,T;
map<string,int>f[maxn];
int b[4][maxn],t[4];
int dfs(int step)
{
	if(f[step].find(T)!=f[step].end())return f[step][T];
	if(step==0)return 0;
	int x=1,y=2,z=3,tot=0;
	if(b[x][t[x]]>b[y][t[y]])swap(x,y);
	if(b[y][t[y]]>b[z][t[z]])swap(y,z);
	if(b[x][t[x]]>b[y][t[y]])swap(x,y);
	
	if(t[x]>0)
	{
		b[y][++t[y]]=b[x][t[x]--];T[b[y][t[y]]-1]=y+'0';tot=(tot+dfs(step-1))%mod;
		b[z][++t[z]]=b[y][t[y]--];T[b[z][t[z]]-1]=z+'0';tot=(tot+dfs(step-1))%mod;
		b[x][++t[x]]=b[z][t[z]--];T[b[x][t[x]]-1]=x+'0';
	}
	if(t[y]>0)
	{
		b[z][++t[z]]=b[y][t[y]--];T[b[z][t[z]]-1]=z+'0';
		tot=(tot+dfs(step-1))%mod;
		b[y][++t[y]]=b[z][t[z]--];T[b[y][t[y]]-1]=y+'0';
	}
	if(T==S)tot++;
	return f[step][T]=tot;
}

int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1,x;i<=n;i++)scanf("%d",&x),S+=x+'0';
	for(int i=1;i<=n;i++)scanf("%d",&a[i]),T+=a[i]+'0';
	b[1][0]=b[2][0]=b[3][0]=999999999;
	for(int i=n;i>=1;i--)b[a[i]][++t[a[i]]]=i;
	f[0][S]=1;
	printf("%d",dfs(m));
}