百度 2021.04.11 笔试 字符串S转化为T的方案

参考牛客大佬：大佬1 大佬2

思路——动态规划

• 记字符串长度为$n$
• 首先将字符串$S=xy$变换为$yx$，等价于将字符串$S$循环右移或左移（读者可以自己举例子尝试一下）
• 那么，只要$T$是$S$的循环同构串，当$S!=T$时，$S$只要经过$1$次操作即可变换为$T$;当$S==T$时，$S$经过$0$次操作即可变换为$T$
• 现在，称字符串循环左移$(j+n)\%n$个位置得到的新字符串称为字符串$j$
• 定义$dp[i][j]$为字符串S经过$i$次操作后，得到字符串$j$的操作方案。其中，$0\le i\le k,0\le j\le n-1$
• 显然字符串$s$经过$1$次操作可以变换到任意字符串$x$，除非$s==x$。同样地，任意字符串$x$也可以经过一次操作变换到字符串$j$，除非$j==x$。
• 那么有
  $$dp[i][j]=sum(dp[i-1][x]),0\le x\le n-1andx\ne j$$
• 然后空间压缩即可。否则$n\ast k$的数据会爆内存
• 时间复杂度$O(nk)$ ，空间复杂度$O(n)$

压缩$i$这个维度的代码：

#include "bits/stdc++.h"
using namespace std;
constexpr int MOD = 1e9 + 7;
using LL = long long;
string S, T;
int k;
LL d[2][1005];
LL sum[2];
int n;
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	while (cin >> S >> T >> k) {
		memset(d, 0, sizeof d);
		memset(sum, 0, sizeof sum);
		n = S.size();
		d[0][0] = 1;
		sum[0] = 1;
		for (int i = 1; i <= k; ++i) {
			sum[i & 1] = 0;
			for (int j = 0; j < n; ++j) {
				d[i & 1][j] = (sum[(i + 1) & 1] + MOD - d[(i + 1) & 1][j]) % MOD;
				sum[i & 1] = (sum[i & 1] + d[i & 1][j]) % MOD;
			}
		}
		LL ans = 0;
		for (int i = 0; i < n; ++i) {
			string tmp = S.substr(i) + S.substr(0, i);
			if (tmp == T) {
				ans = (ans + d[k & 1][i]) % MOD;
			}
		}
		cout << ans << endl;
	}

}
/*
aaca
caaa
2

ac
ca
2
*/

另一种思路

• 首先计算出$S$经过1次变换或0次变换到达$T$的操作方案个数。那么这也是任意一个字符串变换为T的方案个数。记该操作方案个数为$cnt$。注意cnt中可能包含自身变换到自身的情况。
• 定义$dp[i][0]$为经过$i$次操作之后，无法到达$T$的个数。$dp[i][1]$为经过$i$次操作之后，到达$T$的个数
• 那么有
  $$\begin{gathered} dp[i][0]=dp[i-1][0]\times (n-cnt-1)+dp[i-1][1]\times (n-cnt) \\ dp[i][1]=dp[i-1][0]\times cnt+dp[i-1][1]\times (cnt-1) \end{gathered}$$
• 时间复杂度$O(k)$ ，空间复杂度$O(k)$
• 同样可以对第一维的空间压缩，那么空间复杂度将会是$O(1)$！