[蓝桥杯][算法提高VIP]分苹果-差分数组

分苹果

题目描述
小朋友排成一排，老师给他们分苹果。
小朋友从左到右标号1…N。有M个老师，每次第i个老师会给第Li个到第Ri个，一共Ri-Li+1个小朋友每人发Ci个苹果。
最后老师想知道每个小朋友有多少苹果。
数据规模和约定
100%的数据，N、M≤100 000，1≤Li≤Ri≤N，0≤Ci≤100。
输入
第一行两个整数N、M，表示小朋友个数和老师个数。
接下来M行，每行三个整数Li、Ri、Ci，意义如题目表述。
输出
一行N个数，第i个数表示第i个小朋友手上的水果。
样例输入
5 3
1 2 1
2 3 2
2 5 3
样例输出
1 6 5 3 3

1.普通二重循环

#include<bits/stdc++.h>
using namespace std;
int main(){
 int m,n,i,j,l,r,temp;
 int num[100005]={0};
 scanf("%d %d",&m,&n);
 for(i=0;i<n;i++){
  scanf("%d %d %d",&l,&r,&temp);
  for(j=l;j<=r;j++){
   num[j]=num[j]+temp;
  }
 }
 for(i=1;i<=m;i++){
  if(i!=m){
   printf("%d ",num[i]);
  }
  else{
   printf("%d",num[i]);
  }
 }
 getchar();
 getchar();
 return 0;
}

这个代码，用普通的循环 我们得到的时间复杂度为O(m*n)，肯定会超时。

2.差分数组做法

#include<bits/stdc++.h>
using namespace std;
int main() {
 int m, n, i, j, l, r, temp;
 int num[100005] = { 0 };
 scanf("%d %d", &m, &n);
 for (i = 0; i < n; i++) {
  scanf("%d %d %d", &l, &r, &temp);
  num[l] = num[l] + temp;
  num[r + 1] = num[r + 1] - temp;
 }
 for (i = 1; i <= m; i++) {
  num[i] = num[i - 1] + num[i];
  if (i != m) {
   printf("%d ", num[i]);
  }
  else {
   printf("%d", num[i]);
  }
 }
 getchar();
 getchar();
 return 0;
}

所谓差分数组就是指如果两个数组

int d[],a[];
d[1]=a[1];
d[2]=a[2]-a[1];
d[3]=a[3]-a[2];
……
d[i]=a[i]-a[i-1]

此时就称

d[i]是a[i]的差分数组，
d[]i的前缀和就是a[i]，
即a[i]=d[i]+d[i-1]+…+d[0]；

要维护维护差分数组，对于区间[l,r]加temp，我们只需要将num[l]+temp，num[r+1]-temp，这样差分递推时，就可以只在[l,r]加temp。我们这里用的递推来转变为维护一个差分数组就好。^ ^