Bzoj 1901 Dynamic Rankings

【题意】

Description

给定一个含有n个数的序列a[1],a[2],a[3]……a[n],程序必须回答这样的询问:对于给定的i,j,k,在a[i],a[i+1],a[i+2]……a[j]中第k小的数是多少(1≤k≤j-i+1),并且,你可以改变一些a[i]的值,改变后,程序还能针对改变后的a继续回答上面的问题。你需要编一个这样的程序,从输入文件中读入序列a,然后读入一系列的指令,包括询问指令和修改指令。对于每一个询问指令,你必须输出正确的回答。 第一行有两个正整数n(1≤n≤10000),m(1≤m≤10000)。分别表示序列的长度和指令的个数。第二行有n个数,表示a[1],a[2]……a[n],这些数都小于10^9。接下来的m行描述每条指令,每行的格式是下面两种格式中的一种。 Q i j k 或者 C i t Q i j k (i,j,k是数字,1≤i≤j≤n, 1≤k≤j-i+1)表示询问指令,询问a[i],a[i+1]……a[j]中第k小的数。C i t (1≤i≤n,0≤t≤10^9)表示把a[i]改变成为t。

Input

对于每一次询问,你都需要输出他的答案,每一个输出占单独的一行。

Output

Sample Input

5 3
3 2 1 4 7
Q 1 4 3
C 2 6
Q 2 5 3

Sample Output

3
6

HINT

20%的数据中,m,n≤100; 40%的数据中,m,n≤1000; 100%的数据中,m,n≤10000。

【解题方法】动态第K大,主席树套树状数组。

首先静态主席树这个东西其实比较好懂,就是对于每个前缀[1,1],[1,2]....[1,n]都建一颗线段树,将数据离散化后,在线段树中统计区间内所有值出现的次数,每一个线段树都是相同的形态,那么这样我们得到一个很好的性质,这些线段树可以“相减”,假设当前查询的是区间[l,r]内第k大的值,那么我们用前缀[1, r]这个线段树和前缀[1, l-1]这颗线段树通过相减加上二分就可以找到答案。由于相邻两颗线段树最多只有logn个节点不同,我们没有必要完全新建一颗线段树,只需要把相同的结点用指针指一下就行,然后新建logn个结点,这样一来时空复杂度为n*logn。

对于动态查询的主席树,如果修改一个点的值,我们肯定不能修改所有包含这个点的线段树,时间复杂度无法承受。

那么我们就可以套上树状数组,个人感觉这里比较难懂。

树状数组的每个节点都是一颗线段树,但这棵线段树不再保存每个前缀的信息了,而是由树状数组的sum函数计算出这个前缀的信息,那么显而易见这棵线段树保存的是辅助数组S的值,即S=A[i-lowbit+1]+...+A[i],其中A[i]表示值为i的元素出现的次数。

那么对于每次修改,我们要修改树状数组上的logn棵树,对于每棵树,我们要修改logn个结点,所以时空复杂度为

O((n+q)*logn*logn),由于这道题n比较大,查询次数q比较小,所以我们可以初始时建立一颗静态的主席树,树状数组只保存每次修改的信息,那么时空复杂度降为了O(n*logn+q*logn*logn)。

【参考BLOG1】 http://blog.csdn.net/u014664226/article/details/47839973

【参考BLOG2】 http://www.cnblogs.com/kuangbin/p/3308118.html

【AC 代码】

//
//BZOJ 1901
//Created by just_sort 2016/8/22
//All Rights Reserved
//
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 60000+100;
const int maxm = maxn*40;
int n,q,m,tot,a[maxn];
int T[maxn];
vector<int>xs;
int use[maxn],S[maxn];
int getid(int x)
{
    return lower_bound(xs.begin(),xs.end(),x)-xs.begin();
}
struct Q
{
    int l,r,k;
    int ty;
    Q(int ll,int rr,int kk,int ii):l(ll),r(rr),k(kk),ty(ii) {}
    Q() {}
} query[10010];
int lson[maxm],rson[maxm],c[maxm];
int Build(int l,int r)
{
    int root=tot++;
    c[root]=0;
    if(l!=r)
    {
        int mid=(l+r)/2;
        lson[root]=Build(l,mid);
        rson[root]=Build(mid+1,r);
    }
    return root;
}
int Insert(int root,int pos,int val)
{
    int newroot=tot++,tmp=newroot;
    int l=0,r=m-1;
    c[newroot]=c[root]+val;
    while(l<r)
    {
        int mid=(l+r)/2;
        if(pos<=mid)
        {
            lson[newroot]=tot++,rson[newroot]=rson[root];
            newroot=lson[newroot],root=lson[root];
            r=mid;
        }
        else
        {
            lson[newroot]=lson[root],rson[newroot]=tot++;
            newroot=rson[newroot],root=rson[root];
            l=mid+1;
        }
        c[newroot]=c[root]+val;
    }
    return tmp;
}
int lowbit(int x)
{
    return x&-x;
}
void add(int x,int pos,int val)
{
    while(x<=n)
    {
        S[x]=Insert(S[x],pos,val);
        x+=lowbit(x);
    }
}
int getsum(int x)
{
    int ret=0;
    while(x)
    {
        ret+=c[lson[use[x]]];
        x-=lowbit(x);
    }
    return ret;
}

int querykth(int left,int right,int k)
{
    int left_root=T[left-1],right_root=T[right];
    int l=0,r=m-1;
    for(int i=right; i; i-=lowbit(i)) use[i]=S[i];
    for(int i=left-1; i; i-=lowbit(i)) use[i]=S[i];
    while(l<r)
    {
        int mid=(l+r)/2;
        int temp=getsum(right)-getsum(left-1)+c[lson[right_root]]-c[lson[left_root]];
        if(temp>=k){
            r=mid;
            for(int i=left-1; i; i-=lowbit(i)) use[i]=lson[use[i]];
            for(int i=right; i; i-=lowbit(i))  use[i]=lson[use[i]];
            left_root=lson[left_root];
            right_root=lson[right_root];
        }
        else{
            l=mid+1;
            k-=temp;
            for(int i=left-1; i; i-=lowbit(i)) use[i]=rson[use[i]];
            for(int i=right; i; i-=lowbit(i))  use[i]=rson[use[i]];
            left_root=rson[left_root];
            right_root=rson[right_root];
        }
    }
    return l;
}

int main()
{
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        tot=m=0;
        xs.clear();
        for(int i=1; i<=n; i++){
            scanf("%d",&a[i]);
            xs.push_back(a[i]);
        }
        char op[10];
        for(int i=0; i<q; i++){
            scanf("%s",op);
            if(op[0]=='Q'){
                query[i].ty=0;
                scanf("%d%d%d",&query[i].l,&query[i].r,&query[i].k);
            }else{
                query[i].ty=1;
                scanf("%d%d",&query[i].l,&query[i].r);
                xs.push_back(query[i].r);
            }
        }
        sort(xs.begin(),xs.end());
        xs.resize(unique(xs.begin(),xs.end())-xs.begin());
        m = xs.size();
        T[0]=Build(0,m-1);
        for(int i=1; i<=n; i++) T[i]=Insert(T[i-1],getid(a[i]),1);
        for(int i=1; i<=n; i++) S[i]=T[0];
        for(int i=0; i<q; i++){
            if(query[i].ty==0) printf("%d\n",xs[querykth(query[i].l,query[i].r,query[i].k)]);
            else{
                add(query[i].l,getid(a[query[i].l]),-1);
                add(query[i].l,getid(query[i].r),1);
                a[query[i].l]=query[i].r;
            }
        }
    }
    return 0;
}


发布了1454 篇原创文章 · 获赞 213 · 访问量 69万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览