HDU - 3333 Turing Tree(树状数组)

题目大意:给出N个数,M个询问,询问的是[a,b]区间内所有不相同的数的和

解题思路:现在的问题是,怎么处理那些相同的数
我们可以一个个的修改树状数组,边修改,边求值,如果修改的位置刚好是询问的右边界,那就修改后,马上求值,所以要先对问题进行排序,右区间小的先询问
用一个数组纪录上一次相同数出现的地方
现在分两种情况:
一个是,该数之前没有出现过,那就直接修改树状数组,并记录该数所在的位置
另一个是,该数在之前出现过了,那就先将之前的那个数去掉(数组有记录位置),然后再放入该数,再修改该数出现位置,这样就不会出现重复计算的情况了
因为询问的右边界已经确定了,所以肯定会包含当前数的。
询问的区间有可能包含两个相同的数,也有可能只包含一个,考虑如果包含两个的情况,因为我们将前面那个去掉了,所以求得的结果就不会重复加上同一个数了,如果只包含一个的情况,那么包含的数肯定是当前出现的那个。所以去掉前面的数,并不会对后面的求解造成影响

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int Q = 100010;
const int N = 30010;

struct Question{
    int l, r, id;
}q[Q];

LL A[N], B[N], bit[N], last[N], ans[Q];
int n, m;

int cmp(const Question &a, const Question &b) {
    return a.r < b.r;
}

void init() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) {
        scanf("%lld", &A[i]);
        B[i] = A[i];
    }
    scanf("%d", &m);
    for (int i = 1; i <= m; i++) {
        scanf("%d%d", &q[i].l, &q[i].r);
        q[i].id = i;
    }
    sort(B + 1, B + 1 + n);
    sort(q + 1, q + 1 + m, cmp);
}

inline int lowbit(int x) {
    return x & (-x);
}

LL Query(LL x) {
    LL ans = 0;
    while (x) {
        ans += bit[x];
        x -= lowbit(x);
    }
    return ans;
}

void Modify(int x, int w) {
    while (x < N) {
        bit[x] += w;
        x += lowbit(x);
    }
}

int find(int num) {
    int l = 1, r = n;
    while (l <= r) {
        int mid = (l + r) >> 1;
        if (B[mid] >= num) r = mid - 1;
        else l = mid + 1;
    }
    return r + 1;
}

void solve() {
    memset(bit, 0, sizeof(bit));
    memset(last, 0, sizeof(last));
    int cnt = 1;
    for (int i = 1; i <= n; i++) {
        int pos = find(A[i]);
        if (!last[pos]) {
            Modify(i, A[i]);
            last[pos] = i;
        }
        else {
            Modify(last[pos], -A[i]);
            Modify(i, A[i]);
            last[pos] = i;
        }
        while(q[cnt].r == i && cnt <= m) {
            ans[q[cnt].id] = Query(q[cnt].r) - Query(q[cnt].l - 1);
            cnt++;
        }
    }
    for (int i = 1; i <= m; i++)
        printf("%lld\n", ans[i]);

}

int main() {
    int test;
    scanf("%d", &test);
    while (test--) {
        init();
        solve();
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值