洛谷 P3901 数列找不同(“求区间[L,R]中不重复元素的个数”模板)

求区间[L,R]中不重复元素的个数。
思路:可持久化线段树,即对于每个历史版本,仅保留每个元素最靠右的位置。这样,对于每个历史版本有效位置在线段树中被标记为1,无效位置被标记为0,这是一种单点修改+区间查询的基本操作,线段树整体记录区间和。
code:

#include <bits/stdc++.h>

using namespace std;
const int N = 1e5+10 ;
struct Node{
    int l , r , sum ;
};
Node tr[N*40] ;
int tot = 0 ;
int a[N] ;
int ls[N]  ;
int n , Q ;
int root[N] ;
int buildTr(int l , int r)
{
    int p = ++tot ;
    if(l==r)return p ;
    else{
        int mid = (l + r) >> 1 ;
        tr[p].l = buildTr(l,mid) ;
        tr[p].r = buildTr(mid+1,r);
        return p ;
    }
}

int update(int l , int r , int p1 , int val , int pos)
{
    int pp = ++tot ;
    tr[pp] = tr[p1];
    if(l == r){
        tr[pp].sum += val;
        return pp ;
    }else{
        int mid = (l + r) >> 1 ;
        if( pos <= mid ) tr[pp].l = update(l , mid , tr[pp].l , val , pos) ;
        if( pos >= mid + 1) tr[pp].r = update(mid+1 , r , tr[pp].r , val , pos) ;
        tr[pp].sum = tr[tr[pp].l].sum + tr[tr[pp].r].sum ;
        return pp ;
    }
}

int query(int L , int R , int nr , int l , int r)
{
    if(L <= l && r <= R) return tr[nr].sum ;
    else {
        int mid = (l + r) >> 1 ;
        int ans = 0 ;
        if(mid >= L) ans = ans + query(L,R,tr[nr].l,l,mid);
        if(mid+1 <= R) ans = ans + query(L,R,tr[nr].r,mid+1,r) ;
        return ans ;
    }
}

int main()
{
    //freopen("in","r",stdin);
    scanf("%d%d",&n,&Q) ;
    root[0] = buildTr(1,n);
    for(int i = 1 ; i <= n ; i++) {
        scanf("%d",&a[i]);
    }
    for(int i = 1 ; i <= n ; i++) {
        if(ls[a[i]]){
            root[i] = update(1,n,root[i-1],-1,ls[a[i]]);
            root[i] = update(1,n,root[i],1,i) ;
        }else{
            root[i] = update(1,n,root[i-1],1,i);
        }
        ls[a[i]] = i ;
    }
    while(Q--){
        int l , r ;
        scanf("%d%d",&l,&r);
        int len = query(l,r,root[r],1,n);
        if(len == r - l + 1) puts("Yes");
        else puts("No");
    }
    return 0 ;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值