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;
}