Codevs1082线段树练习3

Codevs1082线段树练习3

评测

Codevs1082

题目

【问题大意】
给你N个数,有两种操作:
1:给区间[a,b]的所有数增加X
2:询问区间[a,b]的数的和。
【输入】
第一行一个正整数n,接下来n行n个整数,
再接下来一个正整数Q,每行表示操作的个数,
如果第一个数是1,后接3个正整数,
表示在区间[a,b]内每个数增加X,如果是2,
表示操作2询问区间[a,b]的和是多少。
pascal选手请不要使用readln读入
【输出】
对于每个询问输出一行一个答案
【样例输入】
3
1
2
3
2
1 2 3 2
2 2 3
【样例输出】
9
【数据范围】
(其实这个贴出来没什么用)
1<=n<=200000
1<=q<=200000

分析

用线段树做自然是灰常简单的
假如你想学习如何用树状数组实现可以参考
这篇文章

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
typedef long long lld;
using namespace std;
struct Segment
{
int l,r;
int s1,s2;
lld c;
lld lazy;
}a[400001];
int num=0;
void build(int l,int r)
{
num++;
int t=num;//
a[t].c=0;
a[t].l=l;
a[t].r=r;
a[t].lazy=0;
if(l==r)
{
a[t].s1=-1;
a[t].s2=-1;
return;
}
int mid=(l+r)/2;
a[t].s1=num+1;build(l,mid);
a[t].s2=num+1;build(mid+1,r);
}
void make(int x,int l,int r,int k)
{
a[x].c=a[x].c+k*(r-l+1);
if(a[x].l==l and a[x].r==r)
{
a[x].lazy+=k;
return;
}
int mid=(a[x].r+a[x].l)/2;
if(r<=mid) make(a[x].s1,l,r,k);
else if(l>mid) make(a[x].s2,l,r,k);
else
{
make(a[x].s1,l,mid,k);
make(a[x].s2,mid+1,r,k);
}
}
lld qes(int x,int l,int r)
{
if(a[x].l==l and a[x].r==r) return a[x].c;
int mid=(a[x].r+a[x].l)/2;
int s1=a[x].s1,s2=a[x].s2;
if(a[x].lazy!=0)
{
a[s1].c=a[s1].c+a[x].lazy*(a[s1].r-a[s1].l+1);
a[s2].c=a[s2].c+a[x].lazy*(a[s2].r-a[s2].l+1);
a[s1].lazy+=a[x].lazy;
a[s2].lazy+=a[x].lazy;
a[x].lazy=0;
}
if(r<=mid) return qes(a[x].s1,l,r);
else if(l>mid) return qes(a[x].s2,l,r);
else return qes(a[x].s1,l,mid)+qes(a[x].s2,mid+1,r);
}
int main(int argc, char *argv[])
{
int n,m;
cin>>n;
build(1,n);
for(int i=1;i<=n;i++)
{
int a;
scanf("%d",&a);
make(1,i,i,a);
}
cin>>m;
for(int i=1;i<=m;i++)
{
int x,a,b,c;
scanf("%d %d %d",&x,&a,&b);
if(x==1)
{
scanf("%d",&c);
if(a>b) make(1,b,a,c);
else make(1,a,b,c);
}
else
{
if(a>b) printf("%lld\n",qes(1,b,a));
else printf("%lld\n",qes(1,a,b));
}
}
}

本文基于 知识共享署名-相同方式共享 4.0 国际许可协议发布
本文地址:http://zory.cf/posts/9bea.html
转载请注明出处,谢谢!

哪怕是一杯奶茶,也将鼓励我继续创作!
0%