AtCoder Grand Contest 031 B - Reversi（DP）

B - Reversi

题目链接：https://atcoder.jp/contests/agc031/tasks/agc031_b

题意：

给出n个数，然后现在你可以对一段区间修改成相同的值，前提是左右端点的值相同。问最后这n个数有多少种不同的值。

题解：

设dp[i]表示只考虑1~i这段，有多少不同的值。然后对于当前第i位，有两种选择，修改或者不修改，不修改的话就是dp[i-1]；修改的话就是dp[k]，这里k表示上一个相同颜色的位置。

注意一下如果i-1和i的颜色相同，当前要跳过，这个时候考虑修改是没有意义的。

代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5 + , MOD = 1e9 + ;
int n;
int c[N];
ll dp[N], sum[N];
int main() {
    ios::sync_with_stdio(false);
    cin.tie();
    cin >> n;
    for(int i = ; i <= n; i++)
        cin >> c[i];
    dp[] = ;
    for(int i = ; i <= n; i++) {
        if(c[i] == c[i - ]) {
            dp[i] = dp[i - ];
            continue ;
        }
        dp[i] = (dp[i - ] + sum[c[i]]) % MOD;
        sum[c[i]] = dp[i];
    }
    cout << dp[n];
    return ;
}