码迷,mamicode.com
首页 > 其他好文 > 详细

AtCoder Grand Contest 031 B - Reversi(DP)

时间:2019-03-21 21:49:31      阅读:178      评论:0      收藏:0      [点我收藏+]

标签:pac   text   long   左右   区间   ios   type   include   位置   

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 + 5, MOD = 1e9 + 7;
int n;
int c[N];
ll dp[N], sum[N];
int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> n;
    for(int i = 1; i <= n; i++)
        cin >> c[i];
    dp[0] = 1;
    for(int i = 1; i <= n; i++) {
        if(c[i] == c[i - 1]) {
            dp[i] = dp[i - 1];
            continue ;
        }
        dp[i] = (dp[i - 1] + sum[c[i]]) % MOD;
        sum[c[i]] = dp[i];
    }
    cout << dp[n];
    return 0;
}

 

AtCoder Grand Contest 031 B - Reversi(DP)

标签:pac   text   long   左右   区间   ios   type   include   位置   

原文地址:https://www.cnblogs.com/heyuhhh/p/10574654.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!