计算客模拟赛1、小区划分

小区划分

•  1000ms
•  262144K

一条街道的两侧各连续坐落着 NN 座单元楼。现在要为这些单元楼划分居民校区。

规则如下：

1. 每个小区只能由同一侧连续的若干座单元楼组成。且两侧都恰有 KK 个小区（每个小区至少有一栋楼）。

2. 两侧的小区划分规则应该相同，比如，若左边的房子被分成 {1,2},{3}{1,2},{3} 这两个小区，那么右边也应该如此。

这样两边合计一共有 KK 对小区。

用 a_i,b_ia​i​​,b​i​​ 表示左右两边每座楼的人口在同侧所有单元楼总人口中所占的百分比，定义一个小区的相对拥挤程度为其人口百分比之和（左边就是对应 a_ia​i​​ 的和，右边是对应 b_ib​i​​ 的和)。定义这条街道的总拥挤程度为左右两边 KK 对小区的相对拥挤程度之差的绝对值之和。

现在，请你求出可能的最大拥挤程度。

输入格式

第一行两个整数 NN 和 kk。

第二行 NN 个实数，第 ii 个数为 a_ia​i​​。

第三行 NN 个实数，第 ii 个数位 b_ib​i​​。

输出格式

一个实数，表示这条街道的最大相对拥挤程度，保留到小数点后六位。

数据范围与约定

对于 30\%30% 的数据：n le 20n≤20。

对于 100\%100% 的数据：n le 800,k le 80,k le nn≤800,k≤80,k≤n。

保证 sum a_i=1, sum b_i=1∑a​i​​=1,∑b​i​​=1。

样例解释

11 号楼一个小区，(2,3)(2,3) 号楼 11 个小区。

这样相对拥挤程度最大为 abs(0.1 - 0.4) + abs(0.75 + 0.15 - (0.3 + 0.3))=0.6abs(0.1−0.4)+abs(0.75+0.15−(0.3+0.3))=0.6。

样例输入

3 2
0.1 0.75 0.15
0.4 0.3 0.3

样例输出

0.600000

code

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>

using namespace std;

const int MAXN = 810;
double f[MAXN][85];
double a[MAXN],b[MAXN],sum1[MAXN],sum2[MAXN];

int main() {
    int n,m;
    scanf("%d%d",&n,&m);
    for (int i=1; i<=n; ++i) scanf("%lf",&a[i]),sum1[i] = sum1[i-1]+a[i];
    for (int i=1; i<=n; ++i) scanf("%lf",&b[i]),sum2[i] = sum2[i-1]+b[i];
    
    for (int i=1; i<=n; ++i) {
        f[i][1] = abs(sum1[i]-sum2[i]);
        for (int j=2; j<=m&&j<=i; ++j) {
            for (int k=j-1; k<i; ++k) {
                f[i][j] = max(f[i][j],f[k][j-1]+fabs(sum1[i]-sum1[k]-sum2[i]+sum2[k]));
            }
        }
    }
    printf("%.6lf",f[n][m]);
    return 0;
}