牛客练习赛52 B.Galahad (树状数组)

B.Galahad

链接：https://ac.nowcoder.com/acm/contest/1084/B
来源：牛客网

题目描述

魔女要测试骑士的能力，要求他维护一个长度为 的序列，每次要询问一个区间的和。

但是魔女觉得太简单了，骑士能轻松记住 个数作为前缀和。

于是，魔女要求他回答一个区间的和，但如果某一个数在这个区间出现了多次，这个数只能被计算一次。

输入描述:

第一行两个整数n,m表示数列长度和询问个数。

第二行n个整数，第i个整数为a(i)
(1≤a(i)≤500 000)，表示序列中的第i项。

接下来m行，每行两个整数L,R，表示询问的区间。

输出描述:

输出m行，每行一个整数表示这次询问的答案。

分析：

看到这题的时候感觉见过但是没想起来。
其实就是这道题的变形：求区间内不同数字的个数

每次操作改变量的是a(i)而不是1
具体原理那边写了这里就不写了

ps:
这题莫队tle

code：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
#define ll long long
const int maxm=5e5+5;
struct Node{
    int l,r,id;
    bool operator<(const Node a){
        return r<a.r;
    }
}q[maxm];
int a[maxm];
int mark[maxm];
ll c[maxm],res[maxm];
int n,m;
int lowbit(int i){
    return i&-i;
}
void add(int i,int t){
    while(i<=n){
        c[i]+=t;
        i+=lowbit(i);
    }
}
ll ask(int i){
    ll ans=0;
    while(i){
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
signed main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    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);
    int last=1;
    for(int i=1;i<=m;i++){
        for(int j=last;j<=q[i].r;j++){
            if(mark[a[j]]){
                add(mark[a[j]],-a[j]);
            }
            mark[a[j]]=j;
            add(mark[a[j]],a[j]);
        }
        last=q[i].r+1;
        res[q[i].id]=ask(q[i].r)-ask(q[i].l-1);
    }
    for(int i=1;i<=m;i++){
        printf("%lld\n",res[i]);
    }
    return 0;
}