Codeforces 659G Fence Divercity dp

Fence Divercity

我们设a[ i ] 为第 i 个围栏被切的最靠下的位置， 我们发现a[ i ] 的最大取值有一下信息：

如果从i - 1过来并在 i  结束a[ i ] = min(h[ i - 1], h[ i ])

如果从i - 1过来并延续到i + 1,   a[ i ] = min(h[ i - 1 ], h[ i ], h[ i + 1 ])

如果从 i 开始 i 结束 a[ i ] = h[ i ]

如果从 i 开始并延续到 i + 1,   a[ i ] = min(h[ i ], h[ i + 1])

然后我们就能dp啦， dp[ i ][ 0 ]表示到 i 结束的方案数， dp[ i ][ 1 ]表示到 i 并且要延续下去的方案数。

#include<bits/stdc++.h>
#define LL long long
#define fi first
#define se second
#define mk make_pair
#define PLL pair<LL, LL>
#define PLI pair<LL, int>
#define PII pair<int, int>
#define SZ(x) ((int)x.size())
#define ull unsigned long long

using namespace std;

const int N = 1e6 + ;
const int inf = 0x3f3f3f3f;
const LL INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1e9 + ;
const double eps = 1e-;
const double PI = acos(-);

int n;
LL h[N], dp[N][];

int main() {
    scanf("%d", &n);
    for(int i = ; i <= n; i++) scanf("%lld", &h[i]), h[i]--;
    h[n + ] = inf;
    dp[][] = h[];
    dp[][] = min(h[], h[]);
    LL ans = dp[][];
    for(int i = ; i <= n; i++) {
        dp[i][] = (dp[i - ][] * min(h[i - ], h[i]) + h[i]) % mod;
        dp[i][] = ((dp[i - ][] * min(h[i - ], min(h[i], h[i + ])) % mod) + min(h[i], h[i + ])) % mod;
        ans = (ans + dp[i][]) % mod;
    }
    printf("%lld\n", ans);
    return ;
}

/*
*/