2015南阳CCPC的银牌题。
题意
这里曲折一下原本故事背景。
有n层楼,每层楼只有一个男厕所或者女厕所,现在每层楼的男生有ti人,女生有pi人。
他们每往上走一层或者往下走一层的花费是1个单位。
现在问你应该怎么建厕所才能使总体的花费最少。
思路
不会写啊,怎么推都是一个 O(n3) 的DP,优化不下来。
于是参考了
牛犇的代码思路是这样子的。
dp[第几层楼][所建的性别]
表示,前i层楼,在第i层楼建sex厕所的情况下。最少的代价是多少
我们特别的考虑到,第i层以后的,他们都建跟第i层不一样的厕所。
那么可以得到 ans=min(ans,dp[i][sex]+godown(i,n,sex));这里的godown和goup函数,是特别计算。
goup:性别为sex的人,全都走到l的花费是多少
godown:性别为sex的人全都走到r的花费是多少
那么cnt就是求。l到r区间,这些人去到他们厕所的最短路程。接下来我们每次只要假设。
第j+1层和第i层中间都是同性别的厕所。将他们不同性别的人都走到j或i层即可。
#include <bits/stdc++.h>
#define ll long long
#define MAXN 4005
#define INF 1e18
using namespace std;
struct node {
ll t, p;
} pot[MAXN];
ll sum[MAXN][2], dsum[MAXN][2];
ll dp[MAXN][2];
void init() {
for (int i = 0; i < MAXN; i++) {
dp[i][0] = dp[i][1] = INF;
}
}
void output(ll *num, int len) {
for (int i = 1; i <= len; i++) {
printf("%lld ", num[i]);
}
printf("\n");
}
ll goup(int l, int r, int sex) {
return (sum[r][sex] - sum[l][sex]) * (r + 1) - (dsum[r][sex] - dsum[l][sex]);
}
ll godown(int l, int r, int sex) {
return dsum[r][sex] - dsum[l][sex] - (sum[r][sex] - sum[l][sex]) * l;
}
ll cnt(int l, int r, int sex) {
int mid = l + r >> 1;
return godown(l, mid, sex) + goup(mid, r, sex);
}
int main() {
int T, n, cas = 1;
scanf("%d", &T);
while (T--) {
init();
scanf("%d", &n);
for (int i = 1; i <= n; i++) {
scanf("%lld %lld", &pot[i].t, &pot[i].p);
}
sum[0][0] = sum[0][1] = dsum[0][0] = dsum[0][1] = 0;
for (int i = 1; i <= n; i++) {
sum[i][0] = sum[i - 1][0] + pot[i].t;
sum[i][1] = sum[i - 1][1] + pot[i].p;
dsum[i][0] = dsum[i - 1][0] + pot[i].t * i;
dsum[i][1] = dsum[i - 1][1] + pot[i].p * i;
}
ll ans = INF;
for (int i = 1; i < n; i++) {
dp[i][0] = goup(0, i, 1);
dp[i][1] = goup(0, i, 0);
for (int j = 1; j < i; j++) {
dp[i][0] = min(dp[i][0], dp[j][1] + cnt(j, i, 1));
dp[i][1] = min(dp[i][1], dp[j][0] + cnt(j, i, 0));
}
ans = min(ans, dp[i][0] + godown(i, n, 0));
ans = min(ans, dp[i][1] + godown(i, n, 1));
}
printf("Case #%d: %lld\n", cas++, ans);
}
}
/*
100
4
1 2
2 1
1 2
1 2
3
10 5
4 3
10 5
*/