当前位置: 代码迷 >> 综合 >> HDU 3333 Turing Tree 线段树 or 树状数组
  详细解决方案

HDU 3333 Turing Tree 线段树 or 树状数组

热度:9   发布时间:2024-01-13 18:06:06.0

给出一组序列,问某个区间内序列的和,跟普通求和不同的是,要求值相同的元素只能算一次。

首先看数据范围,就知道要离散化,不然没法标记某个数是否出现过。

然后要对要查询的区间进行排序,按右端点进行排序。

然后按顺序遍历点,如果该点的值以前出现过,就删掉以前的那个,然后插入该点,记录下这个值的位置。刚才已经对询问区间排序了,那么所有右端点在该点的区间都可以进行查询了。

那么为什么要这么干呢。观察一个区间,我们可以发现,如果出现重复的,尽量删除左边的,保留右边的,那么右端点相同的区间都可以进行查询。

下面是树状数组版本的,不得不说树状数组求区间和实在是比较方便


/*
ID: sdj22251
PROG: subset
LANG: C++
*/
#include <iostream>
#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <cmath>
#include <ctime>
#define MAXN 66666
#define MAXM 104444
#define INF 100000000
#define eps 1e-7
#define L(X) X<<1
#define R(X) X<<1|1
using namespace std;
struct interval
{int l, r, id;bool operator <(const interval &a)const{return r < a.r;}
}p[MAXM];
int x[MAXN], a[MAXN], h[MAXN], n;
long long tx[MAXN], ans[MAXM];
int bin(int v, int bound)
{int low = 0, high = bound - 1;while(low <= high){int mid = (low + high) >> 1;if(x[mid] == v) return mid;if(x[mid] < v) low = mid + 1;else high = mid - 1;}return -1;
}
int lowbit(int x)
{return x & -x;
}
void modify(int x, int v)
{for(int i = x; i <= n; i += lowbit(i))tx[i] += v;
}
long long get_sum(int x)
{long long sum = 0;for(int i = x; i > 0; i -= lowbit(i))sum += tx[i];return sum;
}
int main()
{int T, m;scanf("%d", &T);while(T--){scanf("%d", &n);int cnt = 0;for(int i = 1; i <= n; i++){scanf("%d", &a[i]);x[cnt++] = a[i];}memset(tx, 0, sizeof(tx));sort(x, x + cnt);cnt = unique(x, x + cnt) - x;scanf("%d", &m);for(int i = 0; i < m; i++){scanf("%d%d", &p[i].l, &p[i].r);p[i].id = i;}sort(p, p + m);for(int i = 0; i < cnt; i++)h[i] = 0;int j = 0;for(int i = 1; i <= n; i++){int id = bin(a[i], cnt);if(h[id]) modify(h[id], -a[i]);modify(i, a[i]);h[id] = i;for(; j < m; j++){if(p[j].r == i) ans[p[j].id] = get_sum(p[j].r) - get_sum(p[j].l - 1);else break;}}for(int i = 0; i < m; i++)printf("%I64d\n", ans[i]);}return 0;
}


  相关解决方案