2021牛客寒假算法基础集训营1 A-串（dp）

传送门

题意：

长度不超过n，且包含子序列“us”的、只由小写字母构成的字符串有多少个？答案对1e9 + 7取模

思路：

dp推出状态转移方程；

dp[i][0]长度为i没有u;
dp[i][1]长度为i有u无s;
dp[i][2]长度为i有us;

先进行初始化dp[1][0] = 25,dp[1][1] = 1,dp[1][2] = 0;

状态转移方程：
dp[i][0] = dp[i - 1][0] * 25（之前无u，当前位也不选u）

dp[i][1] = dp[i - 1][0]（之前无u，当前位选u） + dp[i - 1][1] * 25（之前已经选了u，当前位不选s）

dp[i][2] = dp[i - 1][2]（之前已有us，当前位随便选） + dp[i - 1][1]（之前有u无s，当前位选s）

因为题目要求长度不超过n，包含“us”子序列，所以dp[i][2]都是满足条件的

代码：

#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <string>
#include <algorithm>
#include <queue>
#include <utility>
#include <stack>
#include <map>
#include <cmath>
#define mes memset 
#define mec memcpy
 
using namespace std;
 
typedef long long ll;
typedef pair<int,int>PII;
 
const int N = 1e6 + 10;
const int null = 0x3f3f3f3f;
const ll mod = 1e9 + 7;

int n; 
ll dp[N][3];

int main()
{
	cin >> n;
	dp[1][0] = 25;//前1个无us的
	dp[1][1] = 1;//前1个有u无s的
	dp[1][2] = 0;//前1个有us的 
	 
	ll ans = 0;
	for(int i = 2;i <= n;i ++)
	{
		dp[i][0] = (dp[i - 1][0] * 25) % mod;
		dp[i][1] = (dp[i - 1][1] * 25 + dp[i - 1][0]) % mod;
		dp[i][2] = (dp[i - 1][1] + dp[i - 1][2] * 26) % mod;
		
		ans = (ans + dp[i][2]) % mod;
	} 
	
	cout << ans << endl;
	
	return 0;
}

一维优化版，降低了空间复杂度（因为每一个状态都是上一个状态得来，只需要保证一维的dp不被污染即可）

#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
#include <string>
#include <algorithm>
#include <queue>
#include <utility>
#include <stack>
#include <map>
#include <cmath>
#define mes memset 
#define mec memcpy
 
using namespace std;
 
typedef long long ll;
typedef pair<int,int>PII;
 
const int N = 1e6 + 10;
const int null = 0x3f3f3f3f;
const ll mod = 1e9 + 7;

int n; 
ll dp[3];

int main()
{
	cin >> n;
	dp[0] = 25;//无us的
	dp[1] = 1;//有u无s的
	dp[2] = 0;//有us的 
	 
	ll ans = 0;
	for(int i = 2;i <= n;i ++)
	{
		dp[2] = (dp[1] + dp[2] * 26) % mod;
		dp[1] = (dp[1] * 25 + dp[0]) % mod;
		dp[0] = (dp[0] * 25) % mod;

		ans = (ans + dp[2]) % mod;
	} 
	
	cout << ans << endl;
	
	return 0;
}