题解/算法 {2967. 使数组成为等数数组的最小代价}

题解/算法 {2967. 使数组成为等数数组的最小代价}

@LINK: https://leetcode.cn/problems/minimum-cost-to-make-array-equalindromic/;

最终整个数组所有元素都要变成T, 题目要求T是回文数, 我们这里先放开这个约束, 那么此时 这个问题 是一个算法模板(中位数-将序列所有元素都变成相同的 最小曼哈顿距离), 即T选择中位数时 代价是最小的;

因此 看一个错误做法, 令T是平均数, 然后找<=T, >=T的第一个回文数, 这是错误的!!! 最小曼哈顿距离 一定是选择中位数;

得到T后, 找<=T, >=T的第一个回文数, 此时 你直接暴力一个个找T, T+1, T+2, ...即可, 因为 比如T是a,b,c,d,e,[?, ?, ?, ?] 你让后半段是[d,c,b,a] 她就是回文的, 也就是前半段abcde 是不会动的, 因此 最多遍历1e5次;

sort( A.begin(), A.end());
auto T = A[ A.size() / 2];
int64_t ANS = Object::INT64_0x3F;
for( auto i = T; i <= int(1e9); ++i){
    auto v = Integer::IntegerToVector( i, -1, 10);
    if( ___Is_palindrome( v.data(), v.size())){
        int64_t sum = 0;
        for( auto j : A){ sum += abs( j - i);}
        ANS = min( ANS, sum);
        break;
    }
}
for( auto i = T; ; --i){
    auto v = Integer::IntegerToVector( i, -1, 10);
    if( ___Is_palindrome( v.data(), v.size())){
        int64_t sum = 0;
        for( auto j : A){ sum += abs( j - i);}
        ANS = min( ANS, sum);
        break;
    }
}
return ANS;