首页 > 代码库 > 喵哈哈村的种花魔法(线段树的区间更新)
喵哈哈村的种花魔法(线段树的区间更新)
喵哈哈村有一个谷歌廖,谷歌廖特别喜欢种花。
而且谷歌廖最神奇的就是,他会施展一种种花魔法,会使得一定区间的花儿,长高k厘米。
在谷歌廖施展若干次魔法之后,好奇的沈宝宝想知道,每朵花儿的高度是多少。
第一行两个整数n,m,分别表示花儿的数量,和谷歌廖施展种花魔法的次数。
第二行n个整数a[i],表示花儿一开始的高度为a[i]厘米。
接下来m行,每行三个整数l,r,k。表示谷歌廖使得区间[l,r]的花儿长高了k厘米。
1<=n,m<=100000
1<=a[i],k<=100000
1<=l<=r<=100000
输出n个整数,即输出每朵花儿在施展魔法之后的高度。
复制
3 1 1 2 3 1 2 5
6 7 3
复制
3 2 1 2 3 1 3 2 1 2 5
8 9 5
这题是线段树区间更新问题
以下贴出常规写法,以及大佬的写法。
1 #include <iostream> 2 #include <cstdio> 3 #include <cstring> 4 #define ll long long 5 #define lson l, m, rt<<1 6 #define rson m+1, r, rt<<1|1 7 const int maxn = 1e6+10; 8 using namespace std; 9 ll flo[maxn<<2]; 10 void PushUP(int rt){ 11 flo[rt] = flo[rt<<1] + flo[rt<<1|1]; 12 } 13 void build(int l, int r, int rt){ 14 if(l == r){ 15 scanf("%lld",&flo[rt]); 16 return; 17 } 18 int m = (l + r) >> 1; 19 build(lson); 20 build(rson); 21 } 22 void update(int L, int R, ll add, int l, int r, int rt){ 23 if(L <= l && r <= R){ 24 flo[rt] += add; 25 return; 26 } 27 int m = (l + r) >> 1; 28 if(m >= L) update(L, R, add, lson); 29 if(m < R) update(L, R, add, rson); 30 //PushUP(rt); 31 } 32 void query(int l, int r, int rt, ll k){ 33 if(l == r){ 34 printf("%lld ",flo[rt]+k); 35 return; 36 } 37 int m = (l + r) >> 1; 38 if(m >= l) query(lson,k+flo[rt]); 39 if(m < r) query(rson,k+flo[rt]); 40 } 41 int main() 42 { 43 int n, m; 44 while(cin>>n>>m){ 45 memset(flo, 0, sizeof(flo)); 46 build(1, n, 1); 47 ll a, b, c; 48 while(m--){ 49 scanf("%d%d%d",&a,&b,&c); 50 update(a, b, c, 1, n, 1); 51 } 52 query(1, n, 1, 0); 53 printf("\n"); 54 } 55 return 0; 56 }
大佬的代码好牛逼,简单易懂。
1 #include <bits/stdc++.h> 2 using namespace std; 3 #define ll long long 4 const int maxn = 1e5+10; 5 ll a[maxn], b[maxn]; 6 int main(){ 7 int n, m; 8 while(cin>>n>>m){ 9 memset(b, 0, sizeof(b)); 10 for(int i = 1; i <= n; i++){ 11 scanf("%lld",&a[i]); 12 } 13 for(int i = 1; i <= m; i++){ 14 ll l, r, val; 15 scanf("%d%d%lld",&l,&r,&val); 16 b[l] += val; 17 b[r+1] -= val; 18 } 19 ll sum = 0; 20 for(int i = 1; i <= n; i++){ 21 sum += b[i]; 22 a[i] += sum; 23 } 24 for(int i = 1; i <= n; i++){ 25 printf("%lld ",a[i]); 26 } 27 printf("\n"); 28 } 29 return 0; 30 }
喵哈哈村的种花魔法(线段树的区间更新)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。