Luogu-P1972 HH的项链

题目链接:https://www.luogu.org/problem/P1972

题目描述:

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。

有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答…… 因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。

输入格式:

一行一个正整数n,表示项链长度。
第二行n个正整数 ai​,表示项链中第i个贝壳的种类。

第三行一个整数 m,表示 HH 询问的个数。

接下来 m行,每行两个整数 l,r,表示询问的区间。

输出格式:

输出m行,每行一个整数,依次表示询问对应的答案。

对于20%的数据,1n,m5000;
对于40%的数据,1n,m105
对于60%的数据,1n,m5×105
对于100%的数据,1n,m,ai1061lrn。

本题解法比较多样,这里放一个离线树状数组的做法:

考虑若干个个r相同的询问,则对于重复出现的数字,只需要关心它的最后一次出现位置,其他出现位置均可以无视。

因此将所有询问按照r排序,再从第一个元素开始,按询问的顺序利用树状数组维护该位置上是否有一个首次出现的数字(如果发现该数字之前出现过,将之前的位置-1,在新位置+1)。

查询[l,r]区间内的答案,可转化为查询前缀和。

#pragma GCC optimize (3)
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
/*#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/hash_policy.hpp>
#include<ext/pb_ds/trie_policy.hpp>
#include<ext/pb_ds/priority_queue.hpp>*/
#define PII pair<int,int>
#define ll long long
#define ull unsigned long long
#define INFI 2147483647
#define INFL 9223372036854775807
#define INFU 18446744073709551615
#define maxn 1000005
using namespace std;
//using namespace __gnu_pbds;
const double PI=acos(-1.0);
const double eps=1e-6;
int n,m,a[maxn],tree[maxn],res[maxn],vis[maxn];
struct Query
{
    int l,r;
    int id;
}q[maxn];
struct Rule
{
    inline bool operator()(const Query& t1,const Query& t2)
    {
        if(t1.r!=t2.r) return t1.r<t2.r;
        if(t1.l!=t2.l) return t1.l<t2.l;
        return t1.id<t2.id;
    }
};
int lowbit(int p)
{
    return p&(-p);
}
void add(int p,int x)
{
    while(p<=n){
        tree[p]+=x;
        p+=lowbit(p);
    }
    return;
}
int query(int p)
{
    int ans=0;
    while(p!=0){
        ans+=tree[p];
        p-=lowbit(p);
    }
    return ans;
}
inline int read()
{
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0' || ch>'9'){
        if(ch=='-') f=-1;
        ch=getchar();
    }
    while(ch>='0' && ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int main()
{
    //ios_base::sync_with_stdio(false);
    //cin.tie(0);
    //cout.tie(0);
    n=read();
    for(int i=1;i<=n;i++) a[i]=read();
    m=read();
    for(int i=1;i<=m;i++){
        q[i].l=read();
        q[i].r=read();
        q[i].id=i;
    }
    sort(q+1,q+1+m,Rule());
    int nxt=1;
    for(int i=1;i<=m;i++){
        for(int j=nxt;j<=q[i].r;j++){
            if(vis[a[j]]) add(vis[a[j]],-1);
                add(j,1);
                vis[a[j]]=j;
        }
        nxt=q[i].r+1;
        res[q[i].id]=query(q[i].r)-query(q[i].l-1);
    }
    for(int i=1;i<=m;i++) printf("%d\n",res[i]);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/HouraisanKaguya/p/11908642.html