codevs 1080~1082 线段树练习系列(模板)

来源:互联网 发布:北外网络教育登录 编辑:程序博客网 时间:2024/06/05 04:30

题目描述 Description
给你N个数,有两种操作:
1:给区间[a,b]的所有数增加X
2:询问区间[a,b]的数的和。

输入描述 Input Description
第一行一个正整数n,接下来n行n个整数,
再接下来一个正整数Q,每行表示操作的个数,
如果第一个数是1,后接3个正整数,
表示在区间[a,b]内每个数增加X,如果是2,
表示操作2询问区间[a,b]的和是多少。
pascal选手请不要使用readln读入

输出描述 Output Description
对于每个询问输出一行一个答案

样例输入 Sample Input
3
1
2
3
2
1 2 3 2
2 2 3

样例输出 Sample Output
9

数据范围及提示 Data Size & Hint
数据范围
1<=n<=200000
1<=q<=200000

题解:这三道题用同一份代码改一下main函数就能AC(这明明就是一道题好吧╭(╯^╰)╮),最简单的线段树模板。

代码如下:

#include<cstdio>#include<cstring>#include<iostream>using namespace std;typedef long long ll;const ll maxn=233333;int xl[maxn];struct jgt{///    int l,r;    ll sum,add;}hah[maxn*4];void build(int p,int l,int r)//建一棵线段树233 {    hah[p].l=l;    hah[p].r=r;    if(l==r)    {        hah[p].sum=xl[l];        return ;    }    int mid=(l+r)/2;    build(p*2,l,mid);    build(p*2+1,mid+1,r);//线段树是一个二叉树233     hah[p].sum=hah[p*2].sum+hah[p*2+1].sum;}void spread(int p)//将标记下放{    if(hah[p].add)    {        hah[p*2].sum+=(hah[p*2].r-hah[p*2].l+1)*hah[p].add;        hah[p*2+1].sum+=(hah[p*2+1].r-hah[p*2+1].l+1)*hah[p].add;        hah[p*2].add+=hah[p].add;        hah[p*2+1].add+=hah[p].add;        hah[p].add=0;    }}ll ask(int p,int l,int r)//区间查询 {    if(l<=hah[p].l&&hah[p].r<=r)    {        return hah[p].sum;    }    spread(p);//下放标记     ll ans=0;    int mid=(hah[p].r+hah[p].l)/2;    if(l<=mid) ans+=ask(p*2,l,r);    if(mid<r) ans+=ask(p*2+1,l,r);    return ans;}void change(int p,int l,int r,ll x)//对于每一次修改我们做一个标记,只有需要的时候才进行计算 {    if(l<=hah[p].l&&hah[p].r<=r)    {        hah[p].sum+=(ll)(hah[p].r-hah[p].l+1)*x;        hah[p].add+=x;        return ;    }    spread(p);//下放标记     int mid=(hah[p].r+hah[p].l)/2;    if(l<=mid) change(p*2,l,r,x);    if(mid<r) change(p*2+1,l,r,x);//继续修改子节点     hah[p].sum=hah[p*2].sum+hah[p*2+1].sum;}int main(){    ll n,q;    scanf("%lld",&n);    for(int i=1;i<=n;i++)        scanf("%d",&xl[i]);    build(1,1,n);//建树     scanf("%lld",&q);    for(int i=1;i<=q;i++)    {        int ss;        scanf("%d",&ss);        if(ss==1)        {            int a,b;            ll x;            scanf("%d%d%lld",&a,&b,&x);            change(1,a,b,x);//修改         }        if(ss==2)        {            int a,b;            scanf("%d%d",&a,&b);            printf("%lld\n",ask(1,a,b));//查询        }    }    return 0;}
2 0
原创粉丝点击