线段树模板题:POJ 3468 a simple problem with integers Java 模板

在这里插入图片描述

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
class Node{
    int left;
    int right;
    long value;
    long add;

}
public class Main {
    static int[] w ;
    static Node[] tree;
    static  BufferedReader br;
    static  int n , m ;
    public static void main(String[] args) throws IOException {
        br = new BufferedReader(new InputStreamReader(System.in));
        String[] s = br.readLine().split(" ");
        n = Integer.parseInt(s[0]);
        m = Integer.parseInt(s[1]);
        w = new int[n+1];
        tree = new Node[n<<2];
        s = br.readLine().split(" ");
        for(int i = 1;  i<= n ; i++) w[i] = Integer.parseInt(s[i-1]);
//        init();
        Build(1,1,n);
        while(m-->0){
            s = br.readLine().split(" ");
            if(s[0].equals("Q")){
                int para1 = Integer.parseInt(s[1]);
                int para2 = Integer.parseInt(s[2]);
                long ans = Query(1,para1,para2);
                System.out.println(ans);
            }else{
                int para1 = Integer.parseInt(s[1]);
                int para2 = Integer.parseInt(s[2]);
                int para3 = Integer.parseInt(s[3]);
                Updata(1,para1,para2,para3);
            }
        }
    }
    private static void Updata(int p, int left, int right, int add) {
        // TODO Auto-generated method stub
        if(tree[p].left==left && tree[p].right==right){
            tree[p].add+=add;
            return;
        }
        tree[p].value += (right-left+1)*add;  //  更新父节点值,不需要再pushup
        int mid=(tree[p].left+tree[p].right)>>1;
        if(right<=mid)
            Updata(p<<1,left,right,add);
        else if(left>=mid+1)
            Updata(p<<1|1,left,right,add);
        else{
            Updata(p<<1,left,mid,add);
            Updata(p<<1|1,mid+1,right,add);
        }
    }
    private static long Query(int p, int left, int right) {
        // TODO Auto-generated method stub
        int len=tree[p].right-tree[p].left+1;
        if(tree[p].left==left && tree[p].right==right)
            return tree[p].value+tree[p].add*len;
        else{
            tree[p<<1].add+=tree[p].add;
            tree[p<<1|1].add+=tree[p].add;
            tree[p].value+=len*tree[p].add;
            tree[p].add=0;
        }
        int mid=(tree[p].left+tree[p].right)>>1;
        if(right<=mid)
            return Query(p<<1,left,right);
        else if(left>=mid+1)
            return Query(p<<1|1,left,right);
        else
            return Query(p<<1,left,mid)+Query(p<<1|1,mid+1,right);
    }
    private static void Build(int p, int left, int right) {
        // TODO Auto-generated method stub
        //System.out.println(p);
        tree[p]=new Node();
        tree[p].left=left;
        tree[p].right=right;
        tree[p].add=0;
        if(left==right){
            tree[p].value=w[left];
            return;
        }
        int mid=(left+right)>>1;
        Build(p<<1,left,mid);
        Build(p<<1|1,mid+1,right);
        tree[p].value=tree[p<<1].value+tree[p<<1|1].value;
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

渝北最后的单纯

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值