您的位置:首页 > 其它

hdu 3874 树状数组

2011-07-25 21:12 369 查看
http://acm.hdu.edu.cn/showproblem.php?pid=3874
来源:
2011 Multi-University Training Contest 4 - Host by SDU
题目:给你一个n(50000)个元素的数组,元素都是非负整数(1000000),然后有q个询问(200000),对每个询问x,y问[x, y]之间的和(注意相同的元素只计算一次)。。。

分析:这题比赛期间始终没做出来,想到了对x排序,然后每次往后走的时候更新l和r的值,这样l只是扫描一遍,但是r却依然是n*n。。。果断杯具
其实这题不难,先对y排序,记录每一个值前一个出现的位置,然后每次一个值v时,若前面出现过v则将前面一个v灭掉,然后将当前位置加上v。。。其实就是保留最后一个出现v的位置的v,前面的地方都归0.。。。唉,很简单的树状数组了。。总是想不出来啊,思想总是跟不上。。。。

。。。一一对应得hash用map<int, int>来做相当方便。。。。

代码:

#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<memory.h>
#include<math.h>
#include<map>
using namespace std;

const int N=50010;
int n, m, a
;
map<int, int> hash;
__int64 ans[N*4], su
;
struct node
{
int i, l, r;
} q[N*4];

void update(int i, int v)
{
while(i<=n)
{
su[i] += v;
i += i&(-i);
}
}

__int64 query(int i)
{
__int64 sum=0;
while(i>0)
{
sum += su[i];
i -= i&(-i);
}
return sum;
}

int cmp1(node &a, node &b)
{
return a.r<b.r;
}

int main()
{
int cas, i, j, k, l, r;
scanf("%d", &cas);
while(cas--)
{
scanf("%d", &n);
for(i=1; i<=n; i++)
scanf("%d", &a[i]);
scanf("%d", &m);
for(i=0; i<m; i++)
{
scanf("%d%d", &q[i].l, &q[i].r);
if(q[i].l>q[i].r)
swap(q[i].l, q[i].r);
q[i].i = i;
}
sort(q, q+m, cmp1);

for(i=0; i<=n; i++)
su[i] = 0;
hash.clear();
r = 1;
for(i=0; i<m; i++)
{
while(r<=q[i].r)
{
if(hash[a[r]]!=0)
update(hash[a[r]], -a[r]);
hash[a[r]] = r;
update(r, a[r]);
r++;
}
ans[q[i].i] = query(q[i].r) - query(q[i].l-1);
}

for(i=0; i<m; i++)
printf("%I64d\n", ans[i]);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: