线段树的区间修改 + 单点修改

时间限制:10000ms

单点时限:1000ms

内存限制: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]中的所有商品的价格之和。

样例输入

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

样例输出

4731
14596


区间更新模板题,记录一下代码.

// #1078 : 线段树的区间修改
#include <iostream>
#include <cstring>
#define N 100005
using namespace std;
int sum[N<<2],lazy[N<<2];

void pushup(int root){
    sum[root] = sum[root<<1] + sum[root<<1|1];
}

void pushdown(int root,int l,int r){
    int mid = (l+r)>>1;
    sum[root<<1] = lazy[root]*(mid-l+1);
    sum[root<<1|1] = lazy[root]*(r-mid);
    lazy[root<<1] = lazy[root];
    lazy[root<<1|1] = lazy[root];
    lazy[root] = -1;
}
void build(int l,int r,int root){
    lazy[root] = -1;
    if(l == r){
        cin>>sum[root];
        return ;
    }
    int mid = (l+r)>>1;
    build(l,mid,root<<1);
    build(mid+1,r,root<<1|1);
    pushup(root);
}

void update(int ll,int rr,int l,int r,int val,int root){
    //l r 为给定修改区间,ll,rr为变换区间
    if(ll>=l&&rr<=r){
        sum[root] = val*(rr-ll+1);
        lazy[root] = val;
        return ;
    }
    int mid = (ll+rr)>>1;
    if(lazy[root]>=0)
        pushdown(root,ll,rr);
    if(mid<r)
        update(mid+1,rr,l,r,val,root<<1|1);
    if(mid>=l)
        update(ll,mid,l,r,val,root<<1);
    pushup(root);
}

int query(int l,int r,int ll,int rr,int root){
    //l,r 为需要查找的区间,ll,rr为变换区间.
    if(l>rr||r<ll){
        return 0;
    }
    if(lazy[root]>=0){
        pushdown(root,ll,rr);
    }
    if(ll>=l&&r>=rr){
        return sum[root];
    }
    int mid = (ll+rr)>>1;
    int cnt = query(l,r,ll,mid,root<<1)+query(l,r,mid+1,rr,root<<1|1);
    return cnt;
}

int n,m;
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin>>n;
    build(1,n,1);
    cin>>m;
    while(m--){
        int k,x,y,z;
        cin>>k;
        if(k == 0){
            cin>>x>>y;
            int ans = query(x,y,1,n,1);
            // cout<<"******"<<endl;
            cout<<ans<<endl;
        }else{
            cin>>x>>y>>z;
            update(1,n,x,y,z,1);
        }
    }
    return 0;
}

I Hate It

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 96242    Accepted Submission(s): 36403

Problem Description

很多学校流行一种比较的习惯。老师们很喜欢询问，从某某到某某当中，分数最高的是多少。
这让很多学生很反感。

不管你喜不喜欢，现在需要你做的是，就是按照老师的要求，写一个程序，模拟老师的询问。当然，老师有时候需要更新某位同学的成绩。

 

Input

本题目包含多组测试，请处理到文件结束。
在每个测试的第一行，有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 )，分别代表学生的数目和操作的数目。
学生ID编号分别从1编到N。
第二行包含N个整数，代表这N个学生的初始成绩，其中第i个数代表ID为i的学生的成绩。
接下来有M行。每一行有一个字符 C (只取'Q'或'U') ，和两个正整数A，B。
当C为'Q'的时候，表示这是一条询问操作，它询问ID从A到B(包括A,B)的学生当中，成绩最高的是多少。
当C为'U'的时候，表示这是一条更新操作，要求把ID为A的学生的成绩更改为B。

 

Output

对于每一次询问操作，在一行里面输出最高成绩。

 

Sample Input

5 6

1 2 3 4 5

Q 1 5

U 3 6

Q 3 4

Q 4 5

U 2 9

Q 1 5

 

Sample Output

5

6

5

9

单点修改的模板题,找最大值.(注意要多组输入)

#include <iostream>
#include <cstring>
using namespace std;
int n,m;
int tree[200005<<2];

void pushup(int rt){
    tree[rt] = max(tree[rt<<1],tree[rt<<1|1]);
}

void build(int l,int r,int rt){
    if(l==r){
        cin>>tree[rt];
        return ;
    }
    int mid = (l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    pushup(rt);
}

void update(int ll,int rr,int rt,int index,int val){
    if(ll==rr&&ll == index){
        tree[rt] = val;
        return ;
    }
    int mid = (ll+rr)>>1;
    if(mid<index)
        update(mid+1,rr,rt<<1|1,index,val);
    else
        update(ll,mid,rt<<1,index,val);
    pushup(rt);
}

int query(int l,int r,int rt,int ll,int rr){
    //rr ll 为修改区间, l r 为给定查询区间
    if(r<ll||l>rr){
        return 0;
    }
    if(l<=ll&&r>=rr){
        return tree[rt];
    }
    int mid = (ll+rr)>>1;
    int cnt = max(query(l,r,rt<<1,ll,mid),query(l,r,rt<<1|1,mid+1,rr));
    return cnt;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    while(cin>>n>>m){
        memset(tree,0,sizeof(tree));
        build(1,n,1);
        while(m--){
            char c;
            int x,y;
            cin>>c;
            cin>>x>>y;
            if(c=='Q'){
                int ans = query(x,y,1,1,n);
                cout<<ans<<endl;
            }else{
                update(1,n,1,x,y);
            }
        }
    }

    return 0;
}