洛谷OJ： P1220 关路灯(DP)

刚看完题目的时候感觉这很容易嘛，不就向左向右两种转移方式，于是直接就写出了动态转移方程，dp[i][j]表示关闭i~j区间内所有灯的最小消耗，dp[i][j] = min{dp[i-1][j]+dis(i,i-1)*w(i-1,j), dp[i][j+1]+dis(j+1,j)*w(i-1,j)}。

后来写出来代码发现不对，思考了一会以后发现没有考虑此时人的位置在哪里，需要多加一种状态。

之后得到了如下方程，dp[i][j][0]表示此时人位于i，dp[i][j][1]表示此时人位于j，那么我们就可以得到真正的动态转移方程了。

dp[i][j][0] = min{dp[i+1][j][0]+dis(i+1,i)*w(i, j+1), dp[i+1][j][1]+dis(j, i)*w(i, j+1)}

dp[i][j][1] = min{dp[i][j-1][0]+dis(j, i)*w(i-1, j), dp[i][j-1][1]+dis(j, j-1)*w(i-1, j) );

那么就可以开始编写代码了，注意dp数组的初始化以及i和j要从哪里开始推。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
using namespace std;

const int maxn = 50+5;
int n, m, dp[maxn][maxn][2], pos[maxn], power[maxn];

int dis(int a, int b) {
    return pos[a]-pos[b];
}
int w(int a, int b) {
    return power[a] + power[n] - power[b-1];
}

int main() {
    cin >> n >> m;
    memset(dp, 0x3f, sizeof(dp));
    dp[m][m][1] = dp[m][m][0] = 0;
    for(int i = 1; i <= n; i++) {
        cin >> pos[i] >> power[i];
        power[i] += power[i-1];
    }
    for(int j = m; j <= n; j++) {
        for(int i = j-1; i >= 1; i--) {
            dp[i][j][0] = min( dp[i+1][j][0] + dis(i+1, i) * w(i, j+1), dp[i+1][j][1] + dis(j, i) * w(i, j+1) );
            dp[i][j][1] = min( dp[i][j-1][0] + dis(j, i) * w(i-1, j), dp[i][j-1][1] + dis(j, j-1) * w(i-1, j) );
        }
    }
    cout << min(dp[1][n][0], dp[1][n][1])<< endl;
    return 0;
}