HDU - 3874 Necklace (线段树 + 离线处理)

Necklace

2
6
1 2 3 4 3 5
3
1 2
3 5
2 6
6
1 1 1 2 3 5
3
1 1
2 4
3 5

3
7
14
1
3
6对于这道题，大家可以看我博客里/article/8817327.html提供的离线处理教程后，基本能够理解离线处理的机制了，然后我们就要对照理解题目，不能出现重复的数字，所以可以离线最右边的值，将前面的值一一删除就可以得到正确答案，当然，大家还需注意的是数值取值范围为long long ，我就是在这个上面看了接近一个小时，query()函数的返回值应该也是long long ，坑爹的地方就是这里了，其他的，大家看代码基本能够秒懂的


#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
typedef long long LL;
#define lson rt << 1, l, mid
#define rson rt << 1|1, mid + 1, r
#define root 1, 1, N
const int MAXN = 5e4 + 5;
const int MAXM = 2e5 + 5;
const int MAXS = 1e6 + 5;
int N, M, T, pre[MAXS];
LL Ans[MAXM],Sum[MAXN << 2];

struct node {
int l, r, id;
bool operator < (const node & object) const {
return r < object.r;
}
} Node[MAXM];

void pushup(int rt) {
Sum[rt] = Sum[rt << 1] + Sum[rt << 1|1];
}

void build(int rt,int l,int r) {
if(l == r) {
scanf("%I64d", &Sum[rt]);
return;
}
int mid = (l + r) >> 1;
build(lson);
build(rson);
pushup(rt);
}

void update(int p,int rt, int l, int r) {
if(l == r) {
Sum[rt] = 0;
return;
}
int mid = (l + r) >> 1;
if(p <= mid) update(p, lson);
else update(p, rson);
pushup(rt);
}

LL query(int L, int R, int rt, int l, int r) {
if(L <= l && r <= R) {
return Sum[rt];
}
int mid = (l + r) >> 1;
LL ret = 0;
if(L <= mid) ret += query(L, R, lson);
if(R > mid) ret += query(L, R, rson);
return ret;
}

int main() {
//freopen("D://imput.txt","r",stdin);
scanf("%d", &T);
while(T --) {
scanf("%d", &N);
build(root);
scanf("%d", &M);
for(int i = 1 ; i <= M ; i ++) {
scanf("%d %d", &Node[i].l, &Node[i].r);
Node[i].id = i;
}
memset(pre, -1, sizeof(pre));
sort(Node + 1, Node + M + 1);
for(int i = 1,j = 1; i <= N; i++) {
int tmp = query(i, i, root);
if(tmp != 0 && pre[tmp] != -1) {//如果前面存在重复的数字则删除他
update(pre[tmp],root);
}
pre[tmp] = i;
while(j <= M && Node[j].r == i) {//如果右边的值等于当前的值，则进行求和，大家可以参考我博客里的教程
Ans[Node[j].id] = query(Node[j].l, Node[j].r, root);
j ++;
}
}
for(int i = 1; i <= M ; i ++) {
printf("%I64d\n", Ans[i]);
}
}
return 0;
}