题目描述
长度不超过nnn,且包含子序列“us”的、只由小写字母构成的字符串有多少个? 答案对109+710^9+7109+7取模。
所谓子序列,指一个字符串删除部分字符(也可以不删)得到的字符串。
例如,“unoacscc"包含子序列"us”,但"scscucu"则不包含子序列"us"
输入描述:
一个正整数nnn(2≤n≤1062 \leq n \leq 10^62≤n≤106)
输出描述:
一个正整数,为满足条件的字符串数量对109+710^9+7109+7取模的值
示例1
输入
2
输出
1
说明
仅有“us”这一个字符串合法
示例2
输入
3
输出
77
说明
长度为3的字符串里,
形状是"u?s"的共有26个
形状是"?us"的共有26个
形状是"us?"的共有26个。
但是,“uss"和"uus"被各多计算了1次,应该减去,
所以共有26*3-2=76个。
再加上长度为2的"us”,所以长度不超过3的合法字符串共有77个。
示例3
输入
874520
输出
16471619
思路:开始时想直接推公式虽然是可行的,但有难度。看了题解才发现自己思维局限性,可以用dp解决因为满足无后效性。设dp[i]为长度为i的合法字符串的数量 那么dp[i + 1] 显然只有两种情况
A:前面i-1有us串了 方案数为dp[i - 1] * 26
B:前面i-1只有u, 那么第i个为s,利用容斥定理 (1~i)的所有组合 - (1~i - 1)中没有u的组合 - (1~i-1)中有us的组合
A + B即为dp[i];
细节见于代码
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e6 + 10;
ll mod = 1e9 + 7;
ll dp[N];
ll n;
ll binarypow(ll a, ll b, ll m) {
ll ans = 1;
while(b > 0)
{
if(b & 1)
ans = ans * a % m;
a = a * a % m;
b >>= 1;
}
return ans;
}
int main()
{
cin >> n;
dp[1] = 0;
dp[2] = 1;
ll ans = 1;
for(int i = 3; i <= n; i++)
{
dp[i] = (dp[i - 1] * 26 + binarypow(26, i - 1, mod) - binarypow(25, i - 1, mod) - dp[i - 1] + mod) % mod;
ans = (ans + dp[i]) % mod;
}
cout << ans << endl;
return 0;
}