description

题目描述
小七擅长茅山道术，有一天他发现了 n 个排成一行的宝石，其中第 i 个宝石
有一个颜色 ci 。
小七决定用茅山道术替换一些宝石，具体地，他每次施法可以选定两个满足
cl = cr 的正整数 l; r(1 ≤ l ≤ r ≤ n) ，然后将区间 [l; r] 内的宝石全部替换为颜
色为 cl 的宝石。
小七想知道，在经过若干次施法后（可以一次也不施法），最终能形成多少
种可能的颜色序列呢？
两个序列不同等价于存在一个 i ，使得位置 i 的宝石在两个序列中颜色不
同。由于答案可能很大，你只需要输出答案对 109 + 7 取模的结果。
输入格式
输入文件 magic.in 包含 2 行。
第 1 行输入一个正整数表示 n 。
第 2 行输入 n 个正整数，其中第 i 个正整数表示 ci 。
输出格式
输出文件 magic.out 包含一行，仅一个非负整数，表示答案对 109 + 7 取模
的结果。
样例输入
6
1 2 1 3 4 3
样例输出
4
数据范围及约定

测试点编号	n	特殊性质
1 ∼ 4	≤ 5	无
5 ∼ 10	≤ 2 × 103	无
11 ∼ 14	≤ 106	ci ≤ 2
15 ∼ 20	≤ 106	无

对于 100% 的数据，满足 1 ≤ k ≤ n ≤ 106; 1 ≤ ci ≤ n

solution

case 1~4 ：随便暴力

case 11~14 ：具有特殊性质$1\le c_i\le 2$

通过找规律（也可以证明），答案为$n$对应的斐波拉契数

case 5~10 ：$n\le 2000$

前面的数据点只是为了不让做不出来的人难看罢了，只有这个数据点是设置的与正解有关的

显然可以承担$n^2$的$dp$，设$f_i:[i,n]$的序列有多少种

暴力枚举$i$可以和哪些$j(i<j)$匹配，即$f_i={\sum }_{j=i+1}^n[c_i=c_j]f_j+1$(不匹配，保持原序列不变)

case 15~20 ：$n\le 10^6,1\le c_i\le n$

想办法在比较暴力的$dp$基础上优化掉一个$n$

发现对于$i$有用的只有和其颜色相同的$j$，每个$i$都会累加后面的所有与之颜色相同的$j$的答案

这其实是个后缀和的形式，后缀和优化，即$g_c:$ 到$i$为止颜色为$c$的所有$j(i<j)$的$f$之和

每次求完$f_i$后，再对$g_{c_i}$进行更新

时间复杂度$O(n)$

还不用对颜色离散化，代码就更简单了

code

#include <cstdio>
#define maxn 1000005
#define int long long
#define mod 1000000007
int n;
int c[maxn], f[maxn], g[maxn];signed main() {freopen( "magic.in", "r", stdin );freopen( "magic.out", "w", stdout );scanf( "%lld", &n );for( int i = 1;i <= n;i ++ ) {scanf( "%lld", &c[i] );if( c[i] == c[i - 1] ) i --, n --;}f[n + 1] = 1;for( int i = n;i;i -- ) {f[i] = ( f[i + 1] + g[c[i]] ) % mod;g[c[i]] = ( g[c[i]] + f[i + 1] ) % mod;}printf( "%lld\n", f[1] );return 0;
}