题解/算法 {100329. 使数组等于目标数组所需的最少操作次数}

题解/算法 {100329. 使数组等于目标数组所需的最少操作次数}

@LINK: https://leetcode.cn/problems/minimum-operations-to-make-array-equal-to-target/;

没想到思路的话 还是很难的…

#看一个错误思路#
我们从A[0]开始 他肯定是需要变成B[0]的, 我们假设d = B[0]-A[0], 即我们此时要执行[0,X] += d 关键是求X, 然后你可能再对[0,X]研究出一个代价值 然后在X = [0,1,2,3,...]里面 找一个最优的;
可是有个问题 如果有若干个X都是最优的 此时要选择哪个呢? 而且代码上很难实现 毕竟N=1e5;

区间所有元素 统一增加/减少 X值, 这是也是个算法模板, 是用差分解决的 (因为差分就是处理区间累加的);
我们令C[i] = A[i]-B[i], 即此时问题转换为: 执行C[l...r] += delta的最小次数 使得C[...] = 0;
令D[]是C[]的差分数组, 那么一次操作定义为: 要么选择D[l]+=d, D[r]-=d(表示[l,...,r-1]区间+=d) 要么选择D[i]+=d(表示[i,i+1,...,N-1] += d), 找到最小操作次数 使得D[...] = 0;

看一个错误思路: 我们从D[0]开始, 对于D[0] 比如他是-5, 那么就在D[>0] 里找一个接近5的 可是现在又有问题 比如你找了一个3 那么此时D[0]=-2 那么你怎么办? 单独的执行两次D[0] += 1吗? 错误 应该再找一个接近2的, 总之很复杂;

正确思路 其实很简单, 就直接贪心就行了 而且这个题不要求方案, 令sum = D[0..,i-1]里面匹配后 剩余的值 然后对于D[i] 让他俩进行消除即可;
. 如果要求方案, 可以对D[]排序 然后首尾指针 相互抵消;

long long minimumOperations(vector<int>& A, vector<int>& T) {
    int N = A.size();
    FOR_( i, 0, N-1){ A[i] -= T[i];}
    FORrev_( i, N-1, 1){ A[i] -= A[i-1];}
    
    long long sum = 0, ANS = 0;
    for( auto i : A){
        if( i>0 && sum<0){
            auto delta = std::min<long long>( i, -sum);
            ANS += delta;
        }
        else if( i<0 && sum>0){
            auto delta = std::min<long long>( -i, sum);
            ANS += delta;
        }
        sum += i;
    }
    return ANS + std::abs(sum);
}