题意:
有个大楼有n层,每层都有两种人,一种喜欢的打球,一种喜欢游泳,现在要你在每一层要么开设球馆,要么开设游泳馆,分配完毕以后,打球的要到球馆,游泳的要到游泳馆,问最优分配下,所有人要移动的最短距离和是多少?
思路:
dp[i][0/1]表示第i层布置为0或1,且i+1层为1或0(与第i层相反)的最小代价。
dp[i][0]可以由dp[j][1]转移过来,其中j小于i,那么我们考虑j+1到i的这些层要建成0,考虑使得代价最小,那么这些层所有1的人都要分别向两边离开。比如3,4,5,6层,3层和4层的1会去2层,5层和6层的1要去7层。
这样状态转移方程为dp[i][0/1] = dp[j][1/0] + tmp。这里tmp的计算可以利用前缀和优化。使得整个算法复杂度为O(n^2)。
另外要注意边界处理,具体实现看代码:
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 4005;
const ll INF = 0x3f3f3f3f3f3f3f3f;
ll a[MAXN][2], dp[MAXN][2], sum[MAXN][2];
int main() {
//freopen("in", "r", stdin);
//freopen("out", "w", stdout);
int T, cs = 0;
scanf("%d", &T);
while (T--) {
int n;
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%lld%lld", &a[i][0], &a[i][1]);
sum[i][0] = sum[i - 1][0] + a[i][0];
sum[i][1] = sum[i - 1][1] + a[i][1];
}
ll ans = INF;
for (int i = 1; i < n; i++) {
for (int k = 0; k < 2; k++) {
dp[i][k] = 0;
for (int j = 1; j <= i; j++)
dp[i][k] += a[j][k ^ 1] * (i - j + 1);
ll tmp = a[i][k ^ 1];
if (i - 1 >= 1) dp[i][k] = min(dp[i][k], dp[i - 1][k ^ 1] + tmp);
for (int j = i - 2; j >= 1; j--) {
int mid = (i - j + 1) / 2 + j;
tmp += sum[mid][k ^ 1] - sum[j][k ^ 1];
dp[i][k] = min(dp[i][k], dp[j][k ^ 1] + tmp);
}
ll tt = 0;
for (int j = i + 1; j <= n; j++) tt += a[j][k] * (j - i);
//cout << tt << endl;
ans = min(ans, dp[i][k] + tt);
}
}
printf("Case #%d: %lld\n", ++cs, ans);
}
return 0;
}