AtCoder Regular Contest 071 F

题面：

给定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;
}