HDU 3874 Necklace(树状数组离线处理）

HDU 3874 Necklace

题目链接

同HDU3333，链接

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;

const int N = 50005;
const int M = 200005;
typedef long long ll;

int t, n, a
;
map<int, int> g;

struct Query {
int l, r, id;
void read(int id) {
scanf("%d%d", &l, &r);
this->id = id;
}
} query[M];

bool cmp(Query a, Query b) {
return a.r < b.r;
}

ll ans[M], bit
;

#define lowbit(x) (x&(-x))

void add(int x, ll v) {
while (x < N) {
bit[x] += v;
x += lowbit(x);
}
}

ll get(int x) {
ll ans = 0;
while (x) {
ans += bit[x];
x -= lowbit(x);
}
return ans;
}

ll get(int l, int r) {
return get(r) - get(l - 1);
}

int main() {
scanf("%d", &t);
while (t--) {
g.clear();
memset(bit, 0, sizeof(bit));
scanf("%d", &n);
for (int i = 1; i <= n; i++)
scanf("%d", &a[i]);
scanf("%d", &n);
for (int i = 0; i < n; i++)
query[i].read(i);
sort(query, query + n, cmp);
int id = 1;
for (int i = 0; i < n; i++) {
while (id <= query[i].r) {
if (g.count(a[id])) {
int v = g[a[id]];
add(v, -a[id]);
add(id, a[id]);
g[a[id]] = id;
} else {
add(id, a[id]);
g[a[id]] = id;
}
id++;
}
ans[query[i].id] = get(query[i].l, query[i].r);
}
for (int i = 0; i < n; i++)
printf("%I64d\n", ans[i]);
}
return 0;
}