Codeforces 295A. Greg and Array(差分)
题目大意其实就是给你一个长度为n的数组,然后依次给了m种类型的操作,其中包
括区间 [l, r] 以及对该区间所加的值val。紧接着又是一个操作区间 [l1, lr],
意思就是我要在mi = l1 --> mi = lr 所有操作执行一遍。最后依次输出这个数组
的最终数值
解题思路:
差分思想
假设arr[] = {0, 0, 0, 0, 0};
然后进行操作 l = 1, r = 3, val = 3;
实际应该是直接操作喽,arr[] -> {0, 3, 3, 3, 0} , 但是一次还好,要是多几次肯定就TimeLimit。
所以就有了差分思想:
即 arr[l] += val, arr[r + 1] -= val;
得 arr[] -> {0, 3, 0, 0, -3};
接下来咱们对数组求前缀和
得 arr[] -> {0, 3, 3, 3, 0};
结果是不是和直接操作结果是一样的呢,而且差分并不能直接操作一次就求出数组,而是直接完成所有的更新操作,最后执行数组还原,这样显然就快了很多,接下来就来看一下代码。
下面展示一些 内联代码片
。
#include "bits/stdc++.h"
#define int long long
using namespace std;
const int N = 1e5 + 10;
struct node{
int l; // 区间左
int r; // 区间右
int val; // 增加值
int cnt; // 该操作执行的次数
};
node opr[N];
signed main() {
int n, m, k;
cin>>n>>m>>k;
vector<int> arr(n + 1);
for(int i = 1; i <= n; i++) cin>>arr[i];
for(int i = 1; i <= m; i++) {
cin>>opr[i].l>>opr[i].r>>opr[i].val;
}
vector<int> b(m + 3, 0);
// 差分其实很简单,你可以试试将它求前缀和即可得到目标数组
for(int i = 1; i <= k; i++) {
int l, r;
cin>>l>>r;
b[l]++;
b[r + 1]--;
}
for(int i = 1; i <= m; i++) {
b[i] += b[i - 1];
opr[i].cnt = b[i];
}
// 到此我们得到了各个操作执行的次数
vector<int> res(n + 2);
for(int i = 1; i <= m; i++) {
res[opr[i].l] += opr[i].val * opr[i].cnt;
res[opr[i].r + 1] -= opr[i].val * opr[i].cnt;
}
for(int i = 1; i <= n; i++) res[i] += res[i - 1];
for(int i = 1; i <= n; i++) {
if(i == 1) cout<<arr[i] + res[i];
else cout<<" "<<arr[i] + res[i];
}
cout<<endl;
return 0;
}