DP训练 玲珑杯线上赛 Round #15 河南专场：A -- Reverse the lights [线性DP]

　又找到原题。

开关灯 (lamp.cpp/c/pas)

【题目描述】
有n个灯，初始时都是不亮的状态，每次你可以选择一个某一个灯，不妨记为x，所有满足和x距离不超过k的灯的状态都将被翻转，选择第i个灯的代价记为c[i]，问最终所有灯都是亮的状态的最小花费。
【输入格式】
输入有两行，第一行包含两个正整数n , k。
第二行包含n个整数，分别表示c[i]。
【输出格式】
一行，表示最小花费。
【样例输入】
3 1
1 1 1
【样例输出】
1
【数据范围】
20%的数据保证：1 <= N <= 20。
100%的数据保证：1 <= N <= 10000 , 0 <= k <= 1000 , 0 <= c[i] <= 1000000000。

思考

就是一个递推式。。。。
令$dp[x]$表示将前x盏灯全部点亮的最小代价，
枚举每个灯的点亮与否进行转移即可。
即$dp[x +2 * K + 1] =min(dp[x + 2 * K + 1], dp[x] + c[x + k + 1])$

#include <bits/stdc++.h>

using namespace std;

const long long inf = 1LL << 61;
const int maxn = 1e4 + 15;

long long dp[maxn];
int n , K , c[maxn];

void up( long long & x , long long v ) { x = min( x , v ) ;}

int main( int argc , char * argv[] ){
    scanf( "%d%d" , & n , & K );
    for(int i = 1 ; i <= n ; ++ i){
        scanf( "%d" , c + i);
        dp[i] = inf;
    }
    for(int i = 1 ; i <= min( n , K + 1 ) ; ++ i) up( dp[ min( n , i + K ) ] , c[i] );
    for(int i = 1 ; i + K + 1 <= n ; ++ i) if( dp[i] != inf ){
        up( dp[min( n , i + 2 * K + 1 )] , dp[i] + c[i + K + 1] );
    }
    cout << dp[n] << endl;
    return 0;
}