ACM新手DAY 18 线段树(2)

题解

A - A Simple Problem with （样题）Integers

题目：有N个整数，A1，A2，…，AN。 您需要处理两种操作。 一种操作是在给定的间隔中为每个数字添加一些给定的数字。 另一种是要求给定间隔中的数字总和。

• 区间合并以及区间更新都有的标准模板。

B - Tunnel Warfare （样题）

• 线段树
• 区间合并求最长连续区间，无区间更新。

C - Just a Hook

A题模板的细节变动，除了输入输出部分，更新操作由加减新的数变成了改变成新的数。

• 注意点：输出部分，最后的句号.也要输出

#include<iostream>
using namespace std;
#define ll long long
const int N=100000;
ll sum[N*4+100];
ll lazy[N*4+100];//lazy是延迟标记
void pushUp(int k)
{
    sum[k]=sum[k*2]+sum[k*2+1];
}
void pushDown(int l,int r,int k)
{
    if(lazy[k]!=0)//不等于0说明有标记，需要向下更新
    {
        int mid=(l+r)/2;
        sum[2*k]=(mid-l+1)*lazy[k];//sum更新部分
        sum[2*k+1]=(r-(mid+1)+1)*lazy[k];

        lazy[2*k] =lazy[k];//lazy的更新部分
        lazy[2*k+1] =lazy[k];
        lazy[k]=0;
    }
}
void build(int l,int r,int k)
{
    lazy[k]=0;
    if(l==r)
    {
        sum[k] = 1;
        return ;
    }
    int mid=(l+r)/2;
    build(l,mid,2*k);
    build(mid+1,r,2*k+1);
    pushUp(k);
}
void update(int l,int r,int ql,int qr,int val,int k)
{
    if(l==ql&&r==qr)
    {
        sum[k] = (r-l+1)*val;
        lazy[k]= val;
        return ;
    }
    pushDown(l,r,k);
    int mid=(l+r)/2;
    if(ql>mid) update(mid+1,r,ql,qr,val,2*k+1);
    else if(qr<=mid) update(l,mid,ql,qr,val,2*k);
    else
    {
        update(l,mid,ql,mid,val,2*k);
        update(mid+1,r,mid+1,qr,val,2*k+1);
    }
    pushUp(k);
}

ll query(int l,int r,int ql,int qr,int k)
{
    if(ql==l&&qr==r)
    {
        return sum[k];
    }
    pushDown(l,r,k);
    int mid=(l+r)/2;
    if(qr<=mid) return query(l,mid,ql,qr,2*k);
    else if(ql>mid) return query(mid+1,r,ql,qr,2*k+1);
    else return query(l,mid,ql,mid,2*k)+query(mid+1,r,mid+1,qr,2*k+1);
}
int main()
{
    ios::sync_with_stdio(0),cin.tie(0);
    int t;
    cin >> t;
    for(int hh = 1; hh <= t; hh++)
    {
        int n,q;
        cin>>n>>q;
        build(1,n,1);//初始化

        int ans;
        for(int i=0; i<q; i++)
        {
            int a,b,c;
            cin>>a>>b>>c;
            update(1,n,a,b,c,1);
        }
        cout << "Case " << hh << ": The total value of the hook is ";
        cout<<query(1,n,1,n,1)<<"."<<endl;
    }
}

E - Color the ball

题目：N个气球排成一排，从左到右依次编号为1,2,3…N.每次给定2个整数a b(a <= b),从气球a开始到气球b依次给每个气球涂一次颜色。算出每个气球被涂过几次颜色

同样是A题的模板变动，不过最后的输出是输出每个气球的涂色次数。

#include <iostream>
#include <string.h>
using namespace std;
#define ll long long

int sum[440000],lazy[440000];

void pushup(int k)
{
    sum[k]=sum[k*2]+sum[k*2+1];
}

void pushdown(int l,int r,int k)
{
    if(lazy[k]!=0)//不等于0说明有标记，需要向下更新
    {
        int mid=(l+r)/2;
        sum[2*k]+=(mid-l+1)*lazy[k];//sum更新部分
        sum[2*k+1]+=(r-(mid+1)+1)*lazy[k];

        lazy[2*k]+=lazy[k];//lazy的更新部分
        lazy[2*k+1]+=lazy[k];
        lazy[k]=0;
    }
}

void BuidTree()
{
    memset(sum,0,sizeof(sum));
    memset(lazy,0,sizeof(lazy));
}

void update(int l,int r,int ql,int qr,int val,int k)
{
    if(l==ql&&r==qr)
    {
        sum[k]+=(r-l+1)*val;
        lazy[k]+=val;
        return ;
    }
    pushdown(l,r,k);
    int mid=(l+r)/2;
    if(ql>mid) update(mid+1,r,ql,qr,val,2*k+1);
    else if(qr<=mid) update(l,mid,ql,qr,val,2*k);
    else
    {
        update(l,mid,ql,mid,val,2*k);
        update(mid+1,r,mid+1,qr,val,2*k+1);
    }
    pushup(k);
}

ll query(int l,int r,int ql,int qr,int k)
{
    if(ql==l&&qr==r)
    {
        return sum[k];
    }
    pushdown(l,r,k);
    int mid=(l+r)/2;
    if(qr<=mid) return query(l,mid,ql,qr,2*k);
    else if(ql>mid) return query(mid+1,r,ql,qr,2*k+1);
    else return query(l,mid,ql,mid,2*k)+query(mid+1,r,mid+1,qr,2*k+1);
}

int main ()
{
    ios::sync_with_stdio(0),cin.tie(0);
    int n,a,b;
    while(cin >> n && n)
    {
        BuidTree();
        for(int i = 1 ; i <= n ; i ++)
        {
            cin >> a >> b;
            update(1,n,a,b,1,1);
        }
        for(int i = 1 ; i <= n ; i ++)
            if(i==n)
                cout << query(1,n,i,i,1) << endl;
            else
                cout << query(1,n,i,i,1) << " ";
    }
    return 0;
}