lamp 开关灯 （dp）

【lamp题解】

10.16

考虑每一盏灯究竟选还是不选，
如果某盏灯x被选择了，那么显然在[x −k,x + k]的灯不应该被选择，
否则这次的选择就是无效的。
于是转换为简单的动态规划问题，
令dp(x)表示将前x盏灯全部点亮的最小代价，
枚举每个灯的点亮与否进行转移即可。
即dp[x +2 * K + 1] =min(dp[x + 2 * K + 1], dp[x] + c[x + k + 1])
复杂度O(n)

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define LL long long
#define N 10010
using namespace std;

int n, k;
LL c[N];
LL f[N], g[N], ans;

int main(){
    freopen ("lamp.in", "r", stdin);
    freopen ("lamp.out", "w", stdout);
    scanf("%d%d", &n, &k);
    ans = (1LL << 60);
    for(int i=1; i<=n; i++){
        scanf("%I64d", &c[i]);
        f[i] = g[i] = ans;
        if(i-k<=1 && i+k>=n) ans = min(ans, c[i]);
    }
    if(k >= n){
        for(int i=1; i<=n; i++){
            ans = min(ans, c[i]);
        }
        cout << ans << endl;
        return 0;
    }
    for(int i=k+1; i<=n; i++){
        if(i-2*k-1 > 0) f[i] = f[i-2*k-1] + c[i-k];
        else f[i] = c[i-k];
    }
    for(int i=n-k; i>=1; i--){
        g[i] = g[i+2*k+1] + c[i+k];
    }
    for(int i=1; i<=n; i++){
        ans = min(ans, f[i] + g[i+1]);
    }
    cout << ans << endl;
}