蓝桥杯算法提高分苹果

最新推荐文章于 2021-08-15 10:37:52 发布

原创最新推荐文章于 2021-08-15 10:37:52 发布 · 3.2k 阅读

0 ·

CC 4.0 BY-SA版权

蓝桥杯专栏收录该内容

32 篇文章

订阅专栏

本文介绍了一种使用线段树解决分配问题的方法，具体场景为老师给排队的小朋友分发苹果。通过算法优化，实现高效且准确地计算每个小朋友分得的苹果数量。

算法提高分苹果

时间限制：1.0s 内存限制：256.0MB

问题描述

　　小朋友排成一排，老师给他们分苹果。
　　小朋友从左到右标号1..N。有M个老师，每次第i个老师会给第Li个到第Ri个，一共Ri-Li+1个小朋友每人发Ci个苹果。
　　最后老师想知道每个小朋友有多少苹果。

输入格式

　　第一行两个整数N、M，表示小朋友个数和老师个数。
　　接下来M行，每行三个整数Li、Ri、Ci，意义如题目表述。

输出格式

　　一行N个数，第i个数表示第i个小朋友手上的水果。

样例输入

样例输出

1 6 5 3 3

数据规模和约定

　　40%的数据，N、M≤1 000。
　　100%的数据，N、M≤100 000，1≤Li≤Ri≤N，0≤Ci≤100。

/*
思路：O(n*m)的纯循环模拟肯定超时  所以用线段树，先初始化所有节点为0
从根节点更新插入的节点，最后结束后再一次性输出并从上到下更新 
*/
#include <iostream>
#include <string.h>
#include <algorithm>
#include <cstdio>
#include <math.h>
using namespace std;
const int N=100000+10;
struct Node{
	int l,r,v;
}node[N*4];
void Init(int now,int le,int ri){    //初始化全部为0 
      if(le==ri){
      	node[now].l=le;
      	node[now].r=ri;
      	node[now].v=0;
      	return;
	  }	
	  int mid=(le+ri)>>1;
	  int ne=now<<1;
	  Init(ne,le,mid);
	  Init(ne+1,mid+1,ri);
	  node[now].l=le;
     node[now].r=ri;
      node[now].v=0;
}
void insert(int now,int le,int ri,int add){        //插入数据 
	if(node[now].l==le&&node[now].r==ri){
		node[now].v+=add;
		return;
	}
    int mid=(node[now].l+node[now].r)>>1;
	int ne=now<<1;
	 if(ri<=mid){
	 	insert(ne,le,ri,add);
	 }    
	else if(le>mid){
		 insert(ne+1,le,ri,add);
	}	     
	else
	{
	    insert(ne,le,mid,add);
	    insert(ne+1,mid+1,ri,add);
	}
} 
void print(int now){
	if(node[now].l==node[now].r){
		cout<<node[now].v<<" ";
		return;
		} 
	int ne=now<<1;
	//父亲的范围大 父亲加的儿子肯定在里面 也要加
	node[ne].v+=node[now].v;
	print(ne);
	node[ne+1].v+=node[now].v;
	print(ne+1);
}
int main(){
    int n,m,i,a,b,c;
    while(cin>>n>>m){
    	Init(1,1,n);
    	for(i=0;i<m;i++){
    		cin>>a>>b>>c;
    		insert(1,a,b,c);
		}
		print(1);
		cout<<endl;
	}
	return 0;
}