【UVA】11992 - Fast Matrix Operations（线段树模板）

标签：style   io   os   for   sp   问题   c   on   amp   

基本的线段树，需要注意的是由于有set和add操作，懒惰标记下推的时候，优先递推set，之后递推add，每次执行set操作将add标记清0

WA了好几次是因为计算那一段的时候出问题了，可笑的是我对着模板找了一个多小时的错。

#include<cstdio>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<algorithm>
using namespace std;
#define lson pos << 1
#define rson pos << 1|1
typedef unsigned long long ULL;
typedef long long LL;
const int maxn = 1111111;
const int  INF = 1 << 30;
struct Node{
    int left,right;
    int max_value,min_value,sum_value;
    int col;
    int col2;
}tr[maxn << 2];
struct Ans{
    int min_value;
    int max_value;
    int sum_value;
    Ans(int a = INF,int b = 0,int c = 0):min_value(a),max_value(b),sum_value(c){};
};
int n,m,k;
void pushdown(int pos){
    if(tr[pos].col){
        int len1 = tr[lson].right - tr[lson].left + 1;
        int len2 = tr[rson].right - tr[rson].left + 1;
        tr[lson].sum_value += (tr[pos].col * len1);
        tr[lson].max_value +=  tr[pos].col;
        tr[lson].min_value +=  tr[pos].col;
        tr[rson].sum_value += (tr[pos].col * len2);
        tr[rson].max_value +=  tr[pos].col;
        tr[rson].min_value +=  tr[pos].col;
        tr[lson].col += tr[pos].col;
        tr[rson].col += tr[pos].col;
        tr[pos].col = 0;
    }
    return ;
}
void pushdown2(int pos){
    if(tr[pos].col2 >= 0){
        int len1 = tr[lson].right - tr[lson].left + 1;
        int len2 = tr[rson].right - tr[rson].left + 1;
        tr[lson].sum_value = (tr[pos].col2 * len1);
        tr[lson].max_value =  tr[pos].col2;
        tr[lson].min_value =  tr[pos].col2;
        tr[rson].sum_value = (tr[pos].col2 * len2);
        tr[rson].max_value =  tr[pos].col2;
        tr[rson].min_value =  tr[pos].col2;
        tr[lson].col2 = tr[pos].col2; tr[lson].col = 0;  //set递推需要清空求和
        tr[rson].col2 = tr[pos].col2; tr[rson].col = 0;
        tr[pos].col2 = -1;
    }
    return ;
}
void pushup(int pos){
    tr[pos].sum_value = tr[lson].sum_value + tr[rson].sum_value;
    tr[pos].max_value = max(tr[lson].max_value,tr[rson].max_value);
    tr[pos].min_value = min(tr[lson].min_value,tr[rson].min_value);
    return ;
}
void BuildTree(int L,int R,int pos){
    tr[pos].left = L; tr[pos].right = R; tr[pos].col = 0;tr[pos].col2 = -1;
    tr[pos].max_value = 0; tr[pos].min_value = 0; tr[pos].sum_value = 0;
    if(L == R) return ;
    int m = (L + R) >> 1;
    BuildTree(L,m,lson);
    BuildTree(m + 1,R,rson);
    return ;
}
void TreeAdd(int l,int r,int add,int pos){
    if(l <= tr[pos].left && tr[pos].right <= r){
        int len = tr[pos].right - tr[pos].left + 1;
        tr[pos].max_value += add; tr[pos].min_value += add;
        tr[pos].sum_value += (add * len); tr[pos].col += add;
        return ;
    }
    pushdown2(pos);
    pushdown(pos);
    int m = (tr[pos].left + tr[pos].right) >> 1;
    if(l <= m)
    TreeAdd(l,r,add,lson);
    if(r  > m)
    TreeAdd(l,r,add,rson);
    pushup(pos);
}
void TreeSet(int l,int r,int value,int pos){
    if(l <= tr[pos].left && tr[pos].right <= r){
        int len = tr[pos].right - tr[pos].left + 1;
        tr[pos].max_value = value; tr[pos].min_value = value;
        tr[pos].sum_value = (value * len); tr[pos].col2 = value;tr[pos].col = 0;
        return ;
    }
    pushdown2(pos);
    pushdown(pos);
    int m = (tr[pos].left + tr[pos].right) >> 1;
    if(l <= m)
    TreeSet(l,r,value,lson);
    if(r  > m)
    TreeSet(l,r,value,rson);
    pushup(pos);
}
Ans Query(int l,int r,int pos){
    if(l <= tr[pos].left && tr[pos].right <= r){
         return Ans(tr[pos].min_value,tr[pos].max_value,tr[pos].sum_value);
    }
    pushdown2(pos);
    pushdown(pos);
    int m = (tr[pos].left + tr[pos].right) >> 1;
    Ans a,b;
    if(l <= m)
        a = Query(l,r,lson);
    if(r  > m)
        b = Query(l,r,rson);
    pushup(pos);
    return Ans(min(a.min_value,b.min_value),max(a.max_value,b.max_value),a.sum_value + b.sum_value);
}
int main(){
    while(scanf("%d%d%d",&n,&m,&k) != EOF){
        BuildTree(1,n * m,1);
        while(k--){
            int t;
            scanf("%d",&t);
            if(t == 1){
                int x1,y1,x2,y2,v;//add
                scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&v);
                for(int i = x1 ; i <= x2 ; i++){
                    int l = (i - 1) * m + y1;
                    int r = (i - 1) * m + y2;
                    TreeAdd(l,r,v,1);
                }
            }
            else if(t == 2){
                int x1,y1,x2,y2,v;//add
                scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&v);
                for(int i = x1 ; i <= x2 ; i++){
                    int l = (i - 1) * m + y1;
                    int r = (i - 1) * m + y2;
                    TreeSet(l,r,v,1);
                }
            }
            else if(t == 3){
                int x1,y1,x2,y2;
                Ans t;
                int MAX = 0,MIN = INF,SUM = 0;
                scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
                for(int i = x1 ; i <= x2 ; i++){
                    int l = (i - 1) * m + y1;
                    int r = (i - 1) * m + y2;
                    t = Query(l,r,1);
                    SUM += t.sum_value;
                    MAX = max(MAX,t.max_value);MIN = min(MIN,t.min_value);
                }
                printf("%d %d %d\n",SUM,MIN,MAX);
            }
        }
    }
    return 0;
}

【UVA】11992 - Fast Matrix Operations（线段树模板）

标签：style   io   os   for   sp   问题   c   on   amp   

原文地址：http://blog.csdn.net/u013451221/article/details/39675149