P3372【模板】线段树1-洛谷-java题解-线段树

传送门：

P3372 【模板】线段树 1 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)https://www.luogu.com.cn/problem/P3372

---

 

---

        这道题是线段树入门的经典模板题，我刚开始学的时候，上网查了很多资料，兜兜转转看了好多个版本，最后把我个人认为比较好的一个模板写在这。初学者一般很难看懂，建议先去看看别人的原理解析。

        注：①对于加减乘除，使用移位运算会更快，比如1<<1|1相当于1*2+1

                ②清除懒人标记可以统一用方法push_down来完成，代码精简化

                ③数据过大，所以我们使用long数组

import java.io.*;
import java.util.*;

public class Main {
    static long[] tree = new long[100000<<2], mark = new long[100000<<2];//mark懒人标记
    static int[] A;//存储数据输入
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StreamTokenizer st = new StreamTokenizer(br);//使用StreamTokenizer快速输入
        PrintWriter pr = new PrintWriter(System.out);
        st.nextToken();int n = (int)st.nval;
        st.nextToken();int m = (int)st.nval;
        A = new int[n+1];
        for(int i=1;i<=n;i++){
            st.nextToken();A[i] = (int)st.nval;
        }
        build(1,n,1);//建树
        for(int i=1;i<=m;i++){
            st.nextToken();int t = (int)st.nval;
            st.nextToken();int x = (int)st.nval;
            st.nextToken();int y = (int)st.nval;
            if(t==1){
                st.nextToken();int k = (int)st.nval;
                update(x,y,k,1,1,n);
            } else {
                pr.println(query(x,y,1,1,n));
            }
        }
        pr.flush();//使用PrintWriter不要忘记flush
    }
    private static void push_down(int p,int l,int r){
        //这里我们使用push_down统一清楚标记
        if(l==r)return;
        int m = l + ((r-l)>>1);
        tree[p<<1] += mark[p]*(m-l+1);
        mark[p<<1] += mark[p];
        tree[p<<1|1] += mark[p]*(r-m);
        mark[p<<1|1] += mark[p];
        mark[p] = 0;
    }
    private static void build(int l,int r,int p){
        if(l==r) {
            tree[p] = A[l];
            return;
        }
        int m = l + ((r-l)>>1);
        build(l,m,p<<1);
        build(m+1,r,p<<1|1);
        tree[p] = tree[p<<1] + tree[p<<1|1];
    }
    private static void update(int l,int r,int d,int p,int ll,int rr){
        if(l<=ll&&rr<=r){
            tree[p] += d * (rr-ll+1);mark[p]+=d;return;
        }
        push_down(p,ll,rr);
        int m = ll + ((rr-ll)>>1);
        if(l<=m) update(l,r,d,p<<1,ll,m);
        if(r>m) update(l,r,d,p<<1|1,m+1,rr);
        tree[p] = tree[p<<1] + tree[p<<1|1];
    }
    private static long query(int l,int r,int p,int ll,int rr){
        if(l<=ll&&rr<=r)return tree[p];
        push_down(p,ll,rr);
        int m = ll + ((rr-ll)>>1);long ans = 0;
        if(l<=m) ans += query(l,r,p<<1,ll,m);
        if(r>m) ans += query(l,r,p<<1|1,m+1,rr);
        return ans;
    }
}