数组操作

Original Link

思想 ：

• 贪心，模拟。
  • 首先对数组进行从小到大排序，再找到第一个 a[idx] != 0 的位置。
  • 对于每次询问，以 base 记录当前数组已经减去的总值，判断时应当计算当前元素与 base 的差值。
  • 若 a[idx] - base > 0 说明需要将后续所有元素减去值 a[idx] - base，将该值累加到 base 中。
  • 若 a[idx] - base == 0 说明当前元素在操作后已经为 \(0\)，则需要向后寻找操作后不为 \(0\) 的元素。
  • 由于贪心的思想，数组中的所有元素一定满足 \(a[idx] - base \le a[idx + 1] - base\)。
  • 最后当 idx 不能再向后移动时，说明全部元素经操作后都变为了 \(0\)。

代码 ：

#include <bits/stdc++.h>
using namespace std;

const int N = 1e6 + 3;

int a[N];

void solve(){
    int n, k; cin >> n >> k;
    for(int i = 0; i < n; i ++) cin >> a[i];
    sort(a, a + n);
    int idx = 0; while(a[idx] == 0 && idx < n) idx ++;  // 找到第一个 a[idx] != 0 的位置
    int base = 0;
    while(k --){
        while(a[idx] - base == 0 && idx < n) idx ++;  // 找到此次询问 a[idx] != 0 的位置
        if(a[idx] - base > 0){
            cout << a[idx] - base << endl;  // 输出经过操作后的值，即 a[idx] - base
            base += a[idx] - base;  // 将剩余的 a[idx] - base 累加到 base 中，后续操作需要减去当前操作的值
        }
        else cout << 0 << endl;
    }
}

int main(){
    solve();
    return 0;
}