hdu 5919 主席树(区间不同数个数+区间第k个数)

题目链接:点击打开链接

题解思路:应用主席树求出区间有多少个不同的数，和区间第k个数(两个功能都考察了)

这里一个技巧就是从后往前插，那么寻找中位数的时间复杂度就变成nlogn了。

#include<bits/stdc++.h>
#define lson l,mid
#define rson mid+1,r
using namespace std;
typedef long long ll;
int n,m;
const int mx =2e5+10;
int num[mx],ans[mx],front[mx],root[mx];
int ls[40*mx],rs[40*mx],sum[40*mx],size;
void update(int x,int &y,int l,int r,int M,int v)
{
    y = ++size;//新区间
    ls[y] = ls[x],rs[y] = rs[x],sum[y] = sum[x] + v;
    //继承上个状态的左右孩子
    int mid = (l+r)>>1;
    if(l==r) return ;//更新一边的儿子
    if(M<=mid) update(ls[x],ls[y],lson,M,v);
    else update(rs[x],rs[y],rson,M,v);
}
int query(int x,int l,int r,int L,int R)
{
    if(L<=l&&R>=r) return sum[x];
    int mid = (l+r)>>1;
    int ans = 0;
    if(L<=mid) ans += query(ls[x],lson,L,R);
    if(R>mid) ans += query(rs[x],rson,L,R);
    return ans;
}
int find_k(int x,int l,int r,int k)
{
    if(l==r) return l;
    int mid = (l+r)>>1;
    if(sum[ls[x]]>=k) return find_k(ls[x],lson,k);
    return find_k(rs[x],rson,k-sum[ls[x]]);
}
int main()
{
    int t,cas = 1;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        memset(front,0,sizeof(front));
        memset(sum,0,sizeof(sum));
        size = 0;
        for(int i=1;i<=n;i++) scanf("%d",num+i);
        for(int i=n;i>=1;i--){
            update(root[i+1],root[i],1,n,i,1);
            //把前一个数删掉
            if(front[num[i]]) update(root[i],root[i],1,n,front[num[i]],-1);
            front[num[i]] = i;
        }
        int l,r;
        for(int i=1;i<=m;i++){
            scanf("%d%d",&l,&r);
            l = (l+ans[i-1])%n+1;
            r = (r+ans[i-1])%n+1;
            if(l>r) swap(l,r);
            int mid = query(root[l],1,n,l,r);//得到区间数的个数
            ans[i] = find_k(root[l],1,n,(mid+1)/2);
            //取中间那个数
        }
        printf("Case #%d:",cas++);
        for(int i=1;i<=m;i++) printf(" %d",ans[i]);
        puts("");
    }
    return 0;
}