题目
小朋友排成一排,老师给他们分苹果。
小朋友从左到右标号1…N。有M个老师,每次第i个老师会给第Li个到第Ri个,一共Ri-Li+1个小朋友每人发Ci个苹果。
最后老师想知道每个小朋友有多少苹果。
输入
第一行两个整数N、M,表示小朋友个数和老师个数。
接下来M行,每行三个整数Li、Ri、Ci,意义如题目表述。
数据规模和约定
100%的数据,N、M≤100 000,1≤Li≤Ri≤N,0≤Ci≤100。
输出
一行N个数,第i个数表示第i个小朋友手上的水果。
样例输入
5 3
1 2 1
2 3 2
2 5 3
样例输出
1 6 5 3 3
解题思路
一开始直接用循环,得分64,时间超限,代码如下:
#include<stdio.h>
int main()
{
int n,m;//N、M,表示小朋友个数和老师个数
int L,R,C,i;
scanf("%d %d\n",&n,&m);
int child[n];
for (i=0;i<n;i++)
child[i] = 0;
while (scanf("%d %d %d",&L,&R,&C)!=EOF){
for (i=L-1;i<R;i++)//1≤Li≤Ri≤N,不需要排序
child[i]+=C;
}
for (i=0;i<n;i++)
{
if (i!=0)
printf(" ");
printf("%d",child[i]);
}
return 0;
}
上述代码的中间部分while嵌套for复杂度太大了,因此,需要考虑如何每一次分苹果不需要把小朋友都遍历一遍,此处采用累加的思路。
(参考:https://blog.csdn.net/hezhiying_/article/details/90380270)
定义一个数组,假如老师分给L到R号小朋友C个苹果,那么对a[L]加上C,为了抵消最后逐项累加计数对未分到的小朋友的苹果数产生影响,因此,对a[R+1]减掉C,那么从R+1号小朋友到末尾的小朋友苹果数都不会受到a[L]值累加过来的影响了。
代码
#include<stdio.h>
int child[100001];//小朋友个数n≤100000
int main()
{
int n,m;//N、M,表示小朋友个数和老师个数
int L,R,C,i;
scanf("%d %d\n",&n,&m);
while (scanf("%d %d %d",&L,&R,&C)!=EOF){
child[L]+=C;
child[R+1]-=C;
}
for (i=1;i<=n;i++)
{
child[i]+=child[i-1];
if (i!=1)
printf(" ");
printf("%d",child[i]);
}
return 0;
}