ZOJ-2112 Dynamic Rankings（分块）

最新推荐文章于 2019-05-11 21:16:11 发布

Paulliant

最新推荐文章于 2019-05-11 21:16:11 发布

阅读量219

点赞数

分类专栏：题目

本文链接：https://blog.csdn.net/Paulliant/article/details/81187116

版权

题目专栏收录该内容

140 篇文章 1 订阅

订阅专栏

题意

有 $n$ 个数， $m$ 个操作，每次操作询问某个区间内第 $k$ 小的值，或改变某个位置上的数，完成这些操作。
$1 \leq n \leq 50000$
$1 \leq m \leq 10000$

思路

询问第 $k$ 小的值难以正面突破时，一般可以二分枚举这个第 $k$ 小的值，取尽量的小的数保证至少有 $k$ 个数小于等于它。同理求第 $k$ 大的值可以取尽量大的数保证至少有 $k$ 个数大于等于它。
仍然是用分块的思想，每次枚举这个第 $k$ 小的数后在询问的区间内找是否有多少个数小于等于它。另开数组，并对完整的块排序，每次查找完整的块是直接在块内 $lower\_bound$ 。而更新时，对一个块重新排序即可。
特别注意的是，这道题可以通过巧妙设计块大小从而优化时间。设块大小为 $S$ ，不难看出，查询不完整块时的复杂度为 $S$ ，而查询完整块的复杂度是 $\dfrac n S \log S$ （ $\log S$ 是 $lower\_bound$ 的复杂度）。那总的复杂度是 $O(S+\dfrac {n\log S } S)$ ，这个式子无法直接用基本不等式，我们将 $\log S$ 变为 $\log n$ （ $\log$ 下差别不大），那复杂度就是 $O(S+\dfrac {n\log n } S)$ ，所以块大小就取 $\sqrt{n\log n }$ 了，复杂度也是 $O(\sqrt{n\log n })$ 级别的。

代码

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
#define N 102003
typedef long long LL;
using namespace std;

template<const int maxn>struct Block
{
    int a[maxn],b[maxn];
    int bs;
    Block(){bs=sqrt(maxn*log2(maxn));}
    int lbound(int L,int R,int x)
    {
        while(L<R)
        {
            int mid=L+R>>1;
            if(b[mid]>=x)
                R=mid;
            else L=mid+1;
        }
        return L;
    }
    int ubound(int L,int R,int x)
    {
        while(L<R)
        {
            int mid=L+R>>1;
            if(b[mid]>x)
                R=mid;
            else L=mid+1;
        }
        return L;
    }
    void build(int *arr,int n)
    {
        memset(b,0x7f,sizeof(b));
        b[0]=0;
        FOR(i,1,n)a[i]=b[i]=arr[i];
        for(int i=0;bs*i<=n;i++)
            sort(b+i*bs,b+i*bs+bs);
        return;
    }
    void update(int x,int val)
    {
        int kx=x/bs;
        b[lbound(kx*bs,kx*bs+bs,a[x])]=val;
        a[x]=val;
        sort(b+kx*bs,b+kx*bs+bs);
        return;
    }
    int func(int x,int y,int f)
    {
        int kx=x/bs,ky=y/bs,res=0;
        if(kx==ky)
        {
            for(int i=x;i<=y;i++)res+=(a[i]<=f);
            return res;
        }
        for(int i=x;i<(kx+1)*bs;i++)
            res+=(a[i]<=f);
        for(int i=ky*bs;i<=y;i++)
            res+=(a[i]<=f);
        FOR(i,kx+1,ky-1)
        {
            int L=i*bs,R=i*bs+bs;
            res+=ubound(L,R,f)-L;
        }
        return res;
    }
    int query(int x,int y,int k)
    {
        int L=0,R=1e9+10;
        while(L<R)
        {
            int mid=L+R>>1;
            if(func(x,y,mid)>=k)
                R=mid;
            else L=mid+1;
        }
        return L;
    }
};
Block<N>B;
int a[N];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        FOR(i,1,n)scanf("%d",&a[i]);
        B.build(a,n);
        FOR(i,1,m)
        {
            char c;
            int x,y,z;
            scanf(" %c",&c);
            if(c=='Q')
            {
                scanf("%d%d%d",&x,&y,&z);
                printf("%d\n",B.query(x,y,z));
            }
            else if(c=='C')
            {
                scanf("%d%d",&x,&y);
                B.update(x,y);
            }
        }
    }
    return 0;
}