zzuli 2900: 阿朱与旭宝宝的高楼游戏

原题链接

样例输入

2
7 3 
4 2 2 1 1 1 2
4 3
1 1 1 2

样例输出

5
1

思路

一定存在一个最高的高楼，这个高度一定是最后的高度，那么我们最先相出的做法就是，找到这个高楼的位置，然后在其两边一直加知道所有的楼都相同，但是这么写特别麻烦，甚至不可写。
对于这种区间操作的问题，我们可以考虑差分，对于差分数组，其中可能会有三种情况 \(> 0 \quad < 0 \quad = 0\) ，对于 \(= 0\) 的情况我们可以忽略。那么只剩下 \(< 0 \quad > 0\),我们先考虑 \(> 0\)的，假设\(a_i > 0\)因为我们只能对连续m个高楼执行 +1 的操作，那么我们需要将 \([i-m,i)\)的区间内都加\(a_i\),这样可以让 i-1的高楼和 i 一致。对于 \(< 0\) 的同理，我们正向做即可。

CODE

点击查看代码

#include <bits/stdc++.h>
using namespace std;
using LL = long long;
const int N = 1e5 + 1;
int a[N];
inline void solve() {
  int n, m; cin >> n >> m;
  for(int i = 1; i <= n; i ++ ) cin >> a[i];
  for(int i = n; i >= 1; i -- ) a[i] -= a[i - 1];
  LL ret = 0;
  for(int i = n; i > m; i -- ) if(a[i] > 0) {
    ret += a[i]; a[i - m] += a[i];
    a[i] = 0; 
  }

  for(int i = 2; i + m - 1 <= n; i ++ ) if(a[i] < 0) {
  	ret -= a[i];
  	a[i + m] += a[i];
  	a[i] = 0;
  }

  for(int i = 2; i <= n; i ++ ) if(a[i] != 0) {
  	ret = -1;
  }
  cout << ret << "\n";
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int T; cin >> T;
  while(T -- ) solve();
 
  return 0;
}