hihoCoder #1078 线段树的区间修改

#1078 : 线段树的区间修改

Time Limit:10000ms

Case Time Limit:1000ms

Memory Limit:256MB

描述

对于小Ho表现出的对线段树的理解，小Hi表示挺满意的，但是满意就够了么？于是小Hi将问题改了改，又出给了小Ho：

假设货架上从左到右摆放了N种商品，并且依次标号为1到N，其中标号为i的商品的价格为Pi。小Hi的每次操作分为两种可能，第一种是修改价格——小Hi给出一段区间[L, R]和一个新的价格NewP，所有标号在这段区间中的商品的价格都变成NewP。第二种操作是询问——小Hi给出一段区间[L, R]，而小Ho要做的便是计算出所有标号在这段区间中的商品的总价格，然后告诉小Hi。

那么这样的一个问题，小Ho该如何解决呢？

提示：推动科学发展的除了人的好奇心之外还有人的懒惰心！

输入

每个测试点（输入文件）有且仅有一组测试数据。

每组测试数据的第1行为一个整数N，意义如前文所述。

每组测试数据的第2行为N个整数，分别描述每种商品的重量，其中第i个整数表示标号为i的商品的重量Pi。

每组测试数据的第3行为一个整数Q，表示小Hi进行的操作数。

每组测试数据的第N+4~N+Q+3行，每行分别描述一次操作，每行的开头均为一个属于0或1的数字，分别表示该行描述一个询问和一次商品的价格的更改两种情况。对于第N+i+3行，如果该行描述一个询问，则接下来为两个整数Li, Ri，表示小Hi询问的一个区间[Li, Ri]；如果该行描述一次商品的价格的更改，则接下来为三个整数Li，Ri，NewP，表示标号在区间[Li, Ri]的商品的价格全部修改为NewP。

对于100%的数据，满足N<=10^5，Q<=10^5, 1<=Li<=Ri<=N，1<=Pi<=N, 0<Pi, NewP<=10^4。

输出

对于每组测试数据，对于每个小Hi的询问，按照在输入中出现的顺序，各输出一行，表示查询的结果：标号在区间[Li, Ri]中的所有商品的价格之和。

Sample Input

10

4733 6570 8363 7391 4511 1433 2281 187 5166 378 

6

1 5 10 1577

1 1 7 3649

0 8 10

0 1 4

1 6 8 157

1 3 4 1557

Sample Output

4731

14596

解题：线段树

 1 #include <iostream>

 2 #include <cstdio>

 3 #include <cstring>

 4 #include <cmath>

 5 #include <algorithm>

 6 #include <climits>

 7 #include <vector>

 8 #include <queue>

 9 #include <cstdlib>

10 #include <string>

11 #include <set>

12 #include <stack>

13 #define LL long long

14 #define pii pair<int,int>

15 #define INF 0x3f3f3f3f

16 using namespace std;

17 const int maxn = 100100;

18 struct node{

19     int lt,rt,lazy,sum;

20 };

21 node tree[maxn<<2];

22 void build(int lt,int rt,int v){

23     tree[v].lt = lt;

24     tree[v].rt = rt;

25     tree[v].lazy = 0;

26     if(lt == rt){

27         scanf("%d",&tree[v].sum);

28         return;

29     }

30     int mid = (lt + rt)>>1;

31     build(lt,mid,v<<1);

32     build(mid+1,rt,v<<1|1);

33     tree[v].sum = tree[v<<1].sum + tree[v<<1|1].sum;

34 }

35 void pushdown(int v){

36     if(tree[v].lazy){

37         tree[v<<1].sum = (tree[v<<1].rt - tree[v<<1].lt + 1)*tree[v].lazy;

38         tree[v<<1|1].sum = (tree[v<<1|1].rt - tree[v<<1|1].lt + 1)*tree[v].lazy;

39         tree[v<<1].lazy = tree[v<<1|1].lazy = tree[v].lazy;

40         tree[v].lazy = 0;

41     }

42 }

43 void update(int lt,int rt,int val,int v){

44     if(tree[v].lt >= lt && tree[v].rt <= rt){

45         tree[v].sum = (tree[v].rt - tree[v].lt + 1)*val;

46         tree[v].lazy = val;

47         return;

48     }

49     pushdown(v);

50     if(lt <= tree[v<<1].rt) update(lt,rt,val,v<<1);

51     if(rt >= tree[v<<1|1].lt) update(lt,rt,val,v<<1|1);

52     tree[v].sum = tree[v<<1].sum + tree[v<<1|1].sum;

53 }

54 int query(int lt,int rt,int v){

55     if(tree[v].lt >= lt && tree[v].rt <= rt)

56         return tree[v].sum;

57     pushdown(v);

58     LL a = 0,b = 0;

59     if(lt <= tree[v<<1].rt) a = query(lt,rt,v<<1);

60     if(rt >= tree[v<<1|1].lt) b = query(lt,rt,v<<1|1);

61     tree[v].sum = tree[v<<1].sum + tree[v<<1|1].sum;

62     return a + b;

63 }

64 int main() {

65     int n,q,op,a,b,val;

66     while(~scanf("%d",&n)){

67         build(1,n,1);

68         scanf("%d",&q);

69         while(q--){

70             scanf("%d %d %d",&op,&a,&b);

71             if(op){

72                 scanf("%d",&val);

73                 update(a,b,val,1);

74             }else printf("%d\n",query(a,b,1));

75         }

76     }

77     return 0;

78 }

View Code