SPOJ DQUERY 区间内不同数的个数

题目:给你n个数,m个询问,问区间[L,R]内的不同数的个数

代码:

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<list>
#include<numeric>
using namespace std;
#define PI acos(-1.0)
#define LL long long
#define ULL unsigned long long
#define INF 0x3f3f3f3f
#define mm(a,b) memset(a,b,sizeof(a))
#define PP puts("*********************");
template<class T> T f_abs(T a){ return a > 0 ? a : -a; }
template<class T> T gcd(T a, T b){ return b ? gcd(b, a%b) : a; }
template<class T> T lcm(T a,T b){return a/gcd(a,b)*b;}
// 0x3f3f3f3f3f3f3f3f
// 0x3f3f3f3f


const int maxn=1e5+50;
const int maxm=maxn*30;
int arr[maxn];//原数组
int root[maxn],lson[maxm],rson[maxm],cnt[maxm];//根节点,左儿子,右儿子,区间内的数出现次数的前缀和
int tot,n;
void init(){
    tot=1;
    root[0]=lson[0]=rson[0]=cnt[0]=0;//建一棵空树
}
int update(int rt,int pos,int val){
    int newrt=tot++,temp=newrt;
    cnt[newrt]=cnt[rt]+val;
    int l=1,r=n;
    while(l<r){
        int mid=(l+r)/2;
        if(pos<=mid){
            r=mid;
            lson[newrt]=tot++;rson[newrt]=rson[rt];
            newrt=lson[newrt];rt=lson[rt];
        }
        else{
            l=mid+1;
            lson[newrt]=lson[rt];rson[newrt]=tot++;
            newrt=rson[newrt];rt=rson[rt];
        }
        cnt[newrt]=cnt[rt]+val;
    }
    return temp;
}
int query(int rt,int pos){
    int l=1,r=n;
    int res=0;
    while(l<r){
        int mid=(l+r)/2;
        if(pos<=mid){
            r=mid;
            res+=cnt[rson[rt]];
            rt=lson[rt];
        }
        else{
            l=mid+1;
            rt=rson[rt];
        }
    }
    res+=cnt[rt];
    return res;
}
map<int,int> mp;
int main(){

    int m,L,R;
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++)
            scanf("%d",&arr[i]);
        init();
        mp.clear();
        for(int i=1;i<=n;i++){
            if(mp.count(arr[i])){
                root[i]=update(root[i-1],mp[arr[i]],-1);
                root[i]=update(root[i],i,1);
            }
            else
                root[i]=update(root[i-1],i,1);
            mp[arr[i]]=i;
        }
        scanf("%d",&m);
        while(m--){
            scanf("%d%d",&L,&R);
            int ans=query(root[R],L);
            printf("%d\n",ans);
        }
    }
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值