HDU 3874 Necklace (树状数组)

来源：互联网发布：zbrush mac 中文编辑：程序博客网时间：2024/05/29 13:24

思路：就是将询问的数据保存下来，按照右区间从小到大排序，因为这样排完序后，可以保证在每次查询的时候，比如 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;}