洛谷 P1972 [SDOI2009]HH的项链 离线+树状数组

https://www.luogu.org/problem/P1972
题目背景

题目描述
HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答……因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。
输入格式
第一行:一个整数N,表示项链的长度。
第二行:N 个整数,表示依次表示项链中贝壳的编号(编号为0 到1000000 之间的整数)。
第三行:一个整数M,表示HH 询问的个数。
接下来M 行:每行两个整数,L 和R(1 ≤ L ≤ R ≤ N),表示询问的区间。
输出格式
M 行,每行一个整数,依次表示询问对应的答案。

思路:把查询按照 r r r从小到大排序,根据贪心的思想,对于若干个询问,若它们的区间右端点是相等的,那么重复出现的元素只记录最右边的那个就可以了。那么就可以用树状数组维护了,如果 a [ i ] a[i] a[i]之前没有出现过, a d d ( i , 1 ) add(i,1) add(i,1)即可,否则既要做 a d d ( i , 1 ) add(i,1) add(i,1),也要消去前一个位置的影响: a d d ( p r e [ a [ i ] ] , − 1 ) add(pre[a[i]],-1) add(pre[a[i]],1)

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define pr pair<int,int>
using namespace std;
typedef long long ll;

const int maxn=1e6+5;

struct query
{
    int id,l,r;
    bool operator <(const query &a)const
    {
        return r<a.r;
    }
}q[maxn];

int pre[maxn];
int tree[maxn];
int a[maxn];
int ans[maxn];
int n,m;

inline int lowbit(int x)
{
    return x&(-x);
}

void add(int i,int v)
{
    for(i;i<=n;i+=lowbit(i))
        tree[i]+=v;
}

int sum(int i)
{
    int ans=0;
    for(;i;i-=lowbit(i))
        ans+=tree[i];
    return ans;
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    scanf("%d",&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);
    int i=1,j=1;
    while(j<=m)
    {
        while(i<=q[j].r)
        {
            add(i,1);
            if(pre[a[i]])
                add(pre[a[i]],-1);
            pre[a[i]]=i;
            ++i;
        }
        ans[q[j].id]=sum(q[j].r)-sum(q[j].l-1);
        ++j;
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值