P1103 书本整理

题目
P1103 书本整理

分析

去掉几本书得到宽度差最小 == 保留就本身得到宽度差最小
所以留下 m = n - k
dp[i][l] 到第i本书截止，一共l本书
i为1dp值肯定为0，一本书就没有宽度差了
所以i和长度l从2开始

代码

#include<bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for (int i = (a); i <= (b); i++) 
int n, k, dp[510][510], m, minn = 1e9;
//dp[i][l]：以i作末尾，选了l本书时的最小花费
struct node {
  int h, w;
}book[1000];
bool cmp(node a, node b) {
  return a.h < b.h;
}
int main() {
  cin >> n >> k;
  m = n-k;
  rep(i, 1, n) cin >> book[i].h >> book[i].w;
  sort(book+1, book+1+n, cmp);
  memset(dp, 20, sizeof(dp));
  rep(i, 1, n) dp[i][1] = 0; 
  rep(i, 2, n) {
    rep (j, 1, i-1) {
      rep(l, 2, min(i, m)) {
        dp[i][l] = min(dp[i][l], dp[j][l-1]+abs(book[i].w-book[j].w));
      }
    }
  }
  for (int i = m; i <= n; i++)
    minn = min(minn, dp[i][m]);
  cout << minn;
  return 0;
}

bye bye lalala