hud 3874 求區間內不同數字的和

http://acm.hdu.edu.cn/showproblem.php?pid=3874

和求區間內不同數字的個數一樣。

//#include<bits/stdc++.h>
//using namespace std;
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=300010;
const int maxq=1000010;
map<int,int>mp;
struct node
{
    int l,r,id;//輸入查詢的區間
    //id記錄的是每個查詢的次序,目的是在對查詢區間排序後,能按原來的查詢順序輸出結果
};
node q[maxq];
bool cmp(node a,node b)
{
    return a.r<b.r;
}
ll c[maxn];
int n;
int lowbit(int x)
{
    return x&(-x);
}

ll sum(ll x)
//int query(int x)
{
   ll res=0;
    while(x)
    {
        res+=c[x];
        x-=lowbit(x);
    }
    return res;
}
void add(ll x,ll val)
{
    while(x<=n)
    {
        c[x]+=val;
        x+=lowbit(x);
    }
}
ll a[maxn],ans[maxq];
int main()
{
    int i,j,cur,Q,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        mp.clear();
        memset(c,0,sizeof(c));
        for(i=1;i<=n;i++)
            scanf("%lld",&a[i]);
        scanf("%d",&Q);
        for(i=1;i<=Q;i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
        }
        sort(q+1,q+1+Q,cmp);//按右端點排序
        cur=1;
        for(i=1;i<=Q;i++)
        {
            for(j=cur;j<=q[i].r;j++)
            {
                if(mp.find(a[j])!=mp.end())//在前面出現過
                {
                    add(mp[a[j]],-a[j]);
                }
                mp[a[j]]=j;
                add(j,a[j]);
            }
            cur=q[i].r+1;
            ans[q[i].id]=sum(q[i].r)-sum(q[i].l-1);
        }
        //一開始cur=1,是1到q[1].r,先對這個小區間操作,然後cur=q[1].r+1
        //是q[1].r到q[2].r,繼續下去
        for(i=1;i<=Q;i++)
            printf("%lld\n",ans[i]);
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章