题面:
给定n,有多少个长度为无穷的数列{a}满足:
1.对于n < k,ak = an
2.对于i < j < k <= i + ai,aj = ak
3.1 <= ai <= n
n <= 10^6,对10^9 + 7取摸
solution:
由限制1,得出真正有效的位置只有n个,所以,定义状态f[i]:长度为i的序列的个数,初值f[1] = n
求解f[i]时,讨论第一个位置所填的数字:
1.在位置1填了数字1,显然对后面的决策没有任何影响,于是变成了递归的子问题,求f[i - 1]即可
2.填了一个大于1的数字k:
i:第二个位置填了数字1,由限制2,一定要连着填k个数字1于是此时有f[i - k - 1]种填法
ii:第二个位置又填了大于1的数字h,那么由限制2,后面所有数字肯定都是h了
iii:如果k较大,第二位又填了1,那么一定一直到末尾都是1了
综上所述,f[i] = f[i - 1] + ∑f[k] (k = 1~i - 3) + (n - 1)^2 + (n - i + 2)
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<stack>
#include<bitset>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;
const int maxn = 1E6 + 10;
typedef long long LL;
const LL mo = 1000000007;
int n,sum,f[maxn];
inline int Mul(const LL &x,const LL &y) {return x * y % mo;}
inline int Add(const int &x,const int &y) {return x + y < mo ? x + y : x + y - mo;}
int main()
{
#ifdef DMC
freopen("DMC.txt","r",stdin);
#endif
cin >> n; f[1] = n; f[2] = Mul(n,n);
for (int i = 3; i <= n; i++)
{
f[i] = Add(f[i - 1],sum);
f[i] = Add(f[i],Mul(n - 1,n - 1));
f[i] = Add(f[i],n - i + 2);
sum = Add(sum,f[i - 2]);
}
cout << f[n] << endl;
return 0;
}