(入门组)寒假每日一题 Week1

1月9日 AcWing 104. 货仓选址

分析

假设有a, b商店, 那么货仓距离每个商店的总距离为$|x-a|+|x-b|$
根据绝对值不等式 $|x-a|+|x-b|>=|a-b|$, 当$x$ 取到[a, b]之间的任何数, 都可以取到使得$|x-a|+|x-b|$取得最小值
同理对于
$$|a_1-x|+|a_2-x|+...|a_n-x|$$
前后一对一对看,
要想使得$|a_1-x|+|a_n-x|$ 取得最小, 那么$x$需要在$[a_1,a_n]$范围内,
取得$|a_k-x|+|a_{k+1}-x|$范围内, $x$必须在$[a_k,a_{k+1}]$内, k为$a_1,a_2,...,a_n$的中位数.
综上,

1. 数组个数为奇数的时候, $x$取数组的中位数
2. 数组个数为偶数的时候, $x$取$[a_k,a_{k+1}]$中的任何一个数

可以取得最小值.
注意: 这里的$a_1,a_2,...,a_n$是排好序的数组, 因此代码中需要先对数组排序

code

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10;
int a[N], n;
int main(){
    cin >> n;
    int res = 0;
    for (int i = 0; i < n; i ++ ) scanf("%d", &a[i]);
    sort(a, a + n);
    for (int i = 0; i < n; i ++ ) res += abs(a[i] - a[n / 2]);
    cout << res << endl;
    return 0;
}

AcWing 898. 数字三角形

分析

如果循环层数i从上往下dp分析, 即:for(int i = 1; i <= n; i ++ ),
那么 f[i][j] = max(f[i - 1][j], f[i][j - 1]) + a[i][j], 需要考虑最左边界的问题

而从下往上面for (int i = n; i >= 1; i --)dp, 就不需要考虑边界问题了, 因为下面的边界囊括上面, 因此上面的状态不会有边界的问题

code(自上到下)

#include <iostream>
#include <cstring>
using namespace std;
const int N = 510;
int a[N][N];
int n;

int main(){
    cin >> n;
    memset(a, -0x3f, sizeof a);
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= i; j ++ )
            cin >> a[i][j];
            
    for (int i = 1; i <= n; i ++ )   
        for (int j = 1; j <= i; j ++ )
            if (i == 1 && j == 1) a[i][j] = a[i][j];
            else if (j == 1) a[i][j] += a[i - 1][j];
            else a[i][j] += max(a[i - 1][j], a[i - 1][j - 1]);
    int res = -1e9;
    for (int i = 1; i <= n; i ++ ) res = max(res, a[n][i]);
    cout << res << endl;
    return 0;
}

code(自下到上, 优化)

#include <iostream>
#include <cstring>
using namespace std;
const int N = 510;
int a[N][N];
int n;

int main(){
    cin >> n;
    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= i; j ++ )
            cin >> a[i][j];
            
    for (int i = n - 1; i >= 1; i --)   
        for (int j = 1; j <= i; j ++ )
            a[i][j] += max(a[i + 1][j], a[i + 1][j + 1]);
    cout << a[1][1] << endl;
    return 0;
}