士兵杀敌(二)

题目:南将军手下有N个士兵,分别编号1到N,这些士兵的杀敌数都是已知的。小工是南将军手下的军师,南将军经常想知道第m号到第n号士兵的总杀敌数,请你帮助小工来回答南将军吧。南将军的某次询问之后士兵i可能又杀敌q人,之后南将军再询问的时候,需要考虑到新增的杀敌数。
思路:可以通过暴力的方法来解决但是算法的复杂度就很高了,也可以采取树状数组来解决,这样可以改善对sum值的修改时间。

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner input=new Scanner(System.in);
        int n=input.nextInt();
        int m=input.nextInt();
        int arr[]=new int[n+1];
        for (int i = 1; i <=n; i++) {
            int t=input.nextInt();
            add(n,i, t, arr);
        }
        String str;
        int index,temp;
        while (m-->0) {
            str=input.next();
            index=input.nextInt();
            temp=input.nextInt();
            if (str.equals("QUERY")) {
                System.out.println(sum(temp, arr)-sum(index-1, arr));
            }else if (str.equals("ADD")) {
                add(n, index, temp, arr);
            }
        }
        // TODO Auto-generated method stub

    }
    public static int lowbit(int x){
        return x&(-x);
    }
    public static int sum(int y,int arr[]){
        int sum=0;
        while (y>0) {
            sum+=arr[y];
            y-=lowbit(y);
        }
        return sum;
    }
    public static void add(int n,int a,int b,int arr[]){
        for (int i = a; i <=n; i+=lowbit(i)) {
            arr[i]+=b;
            a+=lowbit(b);
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值