Necklace【HDU 3874】【树状数组】

  这道题的思路算是比较明白的吧：我们知道有N个数据要输入，以及M次查询，那么我们对于这样序列的N个元素很难有什么处理的方法，不如去退而求其次改变一下输出，我们先给输出对应上编号，然后按照每个询问区间的右端按照升序排列，那么从1～r遍历，只需要是的遍历到的且第二次出现的节点去更新（去掉之前出现的那个节点）即可，我们既然要去掉之前的那个节点，不妨用一个pos[i]标记一下，表示与目前节点相等的前一个节点，但是我们怎么知道前一个节点的位置在哪呢？可以通过再加上一个loc[i]来记录此时所访问到的元素的地址（第几个），然后我们利用pos[i]与loc[i]的关系即可求解。「 pos[i]=loc[val[i]]; loc[val[i]]=i; 」

题目链接

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
using namespace std;
typedef long long ll;
struct node
{
    int l,r,id;
}q[200005];
bool cmp(node e1, node e2)
{
    return e1.r<e2.r;
}
ll N,M;
ll val[50005];     //输入数据
ll pos[50005];     //这个点如果第一次出现，那么它是‘-1’，否则就是先前出现的位置
ll loc[1000005];    //对应值所出现的位置，会随着输入而更新，与pos搭档的作用——初始化为‘-1’
ll bit[400005];     //树状数组的那个数组
ll ans[200005];     //询问、因为是乱序，所以通过ans记录，然后正序输出ans即可
ll Query(int i)
{
    ll sum=0;
    while(i)
    {
        sum+=bit[i];
        i-=lowbit(i);
    }
    return sum;
}
void update(ll i, ll x)
{
    while(i<=N+1)
    {
        bit[i]+=x;
        i+=lowbit(i);
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(val, 0, sizeof(val));        //初始化
        memset(pos, 0, sizeof(pos));
        memset(loc, -1, sizeof(loc));
        memset(bit, 0, sizeof(bit));
        memset(ans, 0, sizeof(ans));
        
        scanf("%lld", &N);
        for(int i=1; i<=N; i++)
        {
            scanf("%lld",&val[i]);
            pos[i]=loc[val[i]];
            loc[val[i]]=i;
            update(i, val[i]);
        }
        
        scanf("%lld",&M);
        for(int i=1; i<=M; i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
        }
        
        sort(q+1, q+1+M, cmp);
        int r=0;
        for(int i=1; i<=M; i++)
        {
            for(int j=r+1; j<=q[i].r; j++)
            {
                if(pos[j]>0)
                {
                    update(pos[j], -val[j]);
                }
            }
            r=q[i].r;
            ans[q[i].id]=Query(q[i].r)-Query(q[i].l-1);
        }
        for(int i=1; i<=M; i++) printf("%lld\n", ans[i]);
    }
    return 0;
}