您的位置:首页 > 其它

HDU 3874 Necklace (树状数组)

2013-03-24 13:58 295 查看
思路:就是将询问的数据保存下来,按照右区间从小到大排序,因为这样排完序后,可以保证在每次查询的时候,比如 1 1 1 2 3 1 这组,我们将重复数字的出现更新在离当前最大查询右区间最近的地方,我们查询[2,6],那么我们就将1更新在6这个位置,这样就能保证每次查询的时候就只加一次1了

还有,这题要注意输出,long long 是必须的,我试了cout<<ans[i]<<endl; TLE;printf("%lld\n",ans[i]);WA 最后用printf("%I64d\n",ans[i]);过了

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=50002;
const int maxm=200002;
typedef long long LL;
LL c[maxn],ans[maxm];
int a[maxn],flag[1000002],n,m;
struct node
{
int l,r,id;
} data[maxm];
inline int lowbit(int x)
{
return x&(-x);
}
void add(int x,LL val)
{
while(x<=n)
c[x]+=val,x+=lowbit(x);
}
LL sum(int x)
{
LL s=0;
while(x>0)
s+=c[x],x-=lowbit(x);
return s;
}
bool cmp(node a,node b)
{
return a.r<b.r;
}
int main()
{
int t,i,num;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
for(i=1; i<=n; i++)
{
scanf("%d",&a[i]);
flag[a[i]]=0;
c[i]=0;
}
scanf("%d",&m);
for(i=0; i<m; i++)
{
scanf("%d%d",&data[i].l,&data[i].r);
data[i].id=i;
}
sort(data,data+m,cmp);
int cnt=1;
for(i=0; i<m; i++)
{
while(cnt<=data[i].r)
{
num=a[cnt];
if(flag[num])
add(flag[num],(LL)-num);
flag[num]=cnt;
add(cnt,(LL)num);
cnt++;
}
ans[data[i].id]=sum(data[i].r)-sum(data[i].l-1);
}
for(i=0; i<m; i++)
printf("%I64d\n",ans[i]);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: