`
java-mans
  • 浏览: 11388789 次
文章分类
社区版块
存档分类
最新评论

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

 
阅读更多

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

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

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

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

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

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


/*
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;
}


分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics