洛谷 P1972 HH的项链(主席树)

思路

  • 用一个 p r e [ i ] pre[i] pre[i] 记录当前位置颜色的前一个相同颜色的位置。
  • 主席树维护 p r e [ i ] pre[i] pre[i]
  • 查找区间 [ 0 , l − 1 ] [0, l-1] [0,l1]

解释

  • 任何一个位置出现了一个颜色,那么产生贡献的区间是 [ 0 , i − 1 ] [0,i-1] [0,i1] 一个颜色没有出现不妨将其上一次出现的位置视为 0 0 0 ,所有在查找区间 [ l , r ] [l,r] [l,r] 内第一次出现的颜色产生的贡献在区间 [ 0 , l − 1 ] [0,l-1] [0,l1] ,所以应该查找 [ 0 , l − 1 ] [0,l-1] [0,l1]
  • 也许有人会问比如我为什么要 r t [ r ] − r t [ l − 1 ] rt[r]-rt[l-1] rt[r]rt[l1] 抽象化主席树的关键操作其实就是简写一下你都知道主席树了肯定知道这说的是啥 ? 因为询问区间 [ l , r ] [l,r] [l,r] 会在 区间 [ 0 , l − 1 ] [0,l-1] [0,l1] 产生贡献,区间 [ 1 , l ] [1,l] [1,l] 也会在区间 [ 0 , l − 1 ] [0,l-1] [0,l1] 产生贡献可能不是这个准确的区间但一定在这个区间内,所以我们要 r t [ r ] − r t [ l − 1 ] rt[r]-rt[l-1] rt[r]rt[l1]
    Think Twice, Code Once
#include<bits/stdc++.h>
#define il inline
#define get getchar
#define put putchar
#define is isdigit
#define re register
#define int long long
#define dfor(i,a,b) for(re int i=a;i<=b;++i)
#define dforr(i,a,b) for(re int i=a;i>=b;--i)
#define dforn(i,a,b) for(re int i=a;i<=b;++i,put(10))
#define mem(a,b) memset(a,b,sizeof a)
#define memc(a,b) memcpy(a,b,sizeof a)
#define pr 114514191981
#define gg(a) cout<<a,put(32)
#define INF 0x7fffffff
#define tt(x) cout<<x<<'\n'
#define ls i<<1
#define rs i<<1|1
#define la(r) tr[r].ch[0]
#define ra(r) tr[r].ch[1]
#define lowbit(x) (x&-x)
using namespace std;
typedef unsigned int ull;
int read(void)
{
    re int x=0,f=1;re char c=get();
    while(!is(c)) (f=c==45?-1:1),c=get();
    while(is(c)) x=(x<<1)+(x<<3)+(c^48),c=get();
    return x*f;
}
void write(int x)
{
    if(x<0) x=-x,put(45);
    if(x>9) write(x/10);
    put((x%10)^48);
}
#define writeln(a) write(a),put(10)
#define writesp(a) write(a),put(32)
#define writessp(a) put(32),write(a)
const int N=1e6+10,M=3e4+10,SN=1e4+10,mod=998244353;
int n,q,cnt,rt[N],pre[N];
struct TR
{
    int l,r,sum;
}tr[N<<5];
int modify(int node,int l,int r,int pos)
{
    tr[++cnt]=tr[node],++tr[cnt].sum,node=cnt;
    if(l==r) return node;
    int mid=(l+r)>>1;
    if(pos<=mid) tr[node].l=modify(tr[node].l,l,mid,pos);
    else tr[node].r=modify(tr[node].r,mid+1,r,pos);
    return node;
}
int query(int u,int v,int l,int r,int x,int y)
{
    if(x<=l&&r<=y) return tr[v].sum-tr[u].sum;
    int mid=(l+r)>>1;
    if(y<=mid) return query(tr[u].l,tr[v].l,l,mid,x,y);
    else if(x>mid) query(tr[u].r,tr[v].r,mid+1,r,x,y);
    else return query(tr[u].l,tr[v].l,l,mid,x,y)+query(tr[u].r,tr[v].r,mid+1,r,x,y);
}
signed main()
{
    n=read();
    re int u,v;
    dfor(i,1,n)
    {
        u=read();
        rt[i]=modify(rt[i-1],0,n,pre[u]);
        pre[u]=i;
    }
    q=read();
    while(q--)
    {
        u=read(),v=read();
        writeln(query(rt[u-1],rt[v],0,n,0,u-1));
    }
    return 0;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Heredy

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值