acwing-273. 分级(《算法竞赛进阶指南》)

acwing-273. 分级(《算法竞赛进阶指南》)

链接跳转

思路:
结论: 每个Bi都是A数组中的某一个

证明:
B数组现在是满足非递减的数列
把A数组排序(从小打大)
反证法: 考虑Bi不是A数组的数
考虑 在 Ai ~ A(i + 1) 之间的 Bj(可能不止一个)，对于这些Bj所对应的Aj, 考虑 Aj < Ai的个数记为x，考虑Aj > A(i + 1)的个数记为y, 将这些Bj 向上或者向下移动d的距离
如果 x < y： 向上移动d, 总答案会 - d(y - x), 有贡献值
x > y: 向下移动d, 总答案会 - d(x - y), 有贡献值
x == y: 怎样移动都行
所以最后肯定会把Bj移动到某一个Ai

现将A排序赋值给C数组
状态表示: f[i][j]: 考虑B数组前i个满足非递减， 且 Bi 为 Cj;
状态转移: 考虑倒数第二个数是 C(k 属于 1 ~ j - 1) 因为 Cj >= Ck
f[i][j] = min(f[i - 1][k] && k < j) + abs(a[i] - c[j])（代价）;
现在时间复杂度O(N ^ 3)
可以通过前缀最小值来减少时间复杂度
O(N^2)

求非递增时: 将A数组反过来 考虑非递减

代码:

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 2010;

int f[N][N];
int a[N], c[N];
int n;

int work(){
    for(int i = 1; i <= n; ++ i) c[i] = a[i];
    sort(c + 1, c + n + 1);
    
    for(int i = 1; i <= n; ++ i){
        int mn = 2e9 + 10;
        for(int j = 1; j <= n; ++ j){
            mn = min(mn, f[i - 1][j]);
            f[i][j] = mn + abs(c[j] - a[i]);
        }
    }
    
    int mn = 2e9 + 10;
    
    for(int i = 1; i <= n; ++ i) mn = min(mn, f[n][i]);
    
    return mn;
}

int main(){
    cin >> n;
    for(int i = 1; i <= n; ++ i) cin >> a[i];
    int mn = work();
    reverse(a + 1, a + n + 1);
    mn = min(mn, work());
    cout << mn << endl;
    return 0;
}