AcWing Problem - 一个简单的整数问题

5 篇文章 0 订阅

AcWing Problem - 一个简单的整数问题

原题地址

题目类型:树状数组
题意

有两种操作:
• 求数组中一段区间的和
• 修改数组中某一个数

分析

使用树状数组和差分的思修来维护。时间复杂度 n log⁡n

代码
static int[] C;
static int[] a;

public static void solve() throws IOException {
    int n = nextInt();
    int m = nextInt();
    
    init(n);
    
    for (int i = 1; i <= n; i++) a[i] = nextInt();
    for (int i = 1; i <= n; i++) add(i, a[i] - a[i - 1], n);
    
    while (m-- > 0) {
        char op = nextChar();
        
        if (op == 'Q') {
            int x = nextInt();
            pw.println(query(x));
        } else {
            int l = nextInt();
            int r = nextInt();
            int v = nextInt();
            add(l, v, n);
            add(r + 1, -v, n);
        }
    }
} 

public static void add(int x, int v, int n) {
    while (x <= n) {
        C[x] += v;
        x += lowbit(x);
    }
}

public static long query(int x) {
    long re = 0;
    while (x > 0) {
        re += C[x];
        x -= lowbit(x);
    }
    return re;
}

public static int lowbit(int x) { return x &(-x); }

public static void init(int n) {
    a = new int[n + 1];
    C = new int[n + 1];
}

/*******************************************************************************************/


/********************************************************************************************/

// Fast I/O

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值