hdu3727Jewel(劃分樹+二分)

題目請戳這裏

題目大意:給許多操作,4類:

1,insert x,插入一個數x,x範圍2^31內(題目描述有誤)

2,query_1 s t k,查詢範圍[s,t]內第k小的數,s<=t<=len(當前數字長度)

3,query_2 x   查詢x的rank,從小到大。x保證出現過

4,query_3 x,查詢[1,len]第x小的數。

輸出3個數,爲3個詢問累加和。

題目分析:由於沒有刪除操作,所以加入的數字序列是不變的,所以離線處理。

q1和q3其實是一個操作,查詢區間第k大值問題,劃分樹搞。

q2的話,其實也可以通過二分轉化爲劃分樹操作。如果願意,寫棵線段樹單獨維護一下也是可以的,不過要離散化。爲了偷懶,選擇二分+劃分樹啦

關於劃分樹,其實也是基於線段樹的一個數據結構,模擬歸併排序的過程,不斷分治。詳細講解戳這裏

詳情請見代碼:

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 100005;
const int M = 110005;
const int DEEP = 20;
typedef __int64 ll;
ll ans[3];
struct node
{
    int optype;
    int a,b,c;
}op[M];
int n,m;
char str[20];
int num[DEEP][N],lefta[DEEP][N],lcm[N];

void build(int s,int e,int nm,int dp)
{
    if(s == e)
        return;
    int mid = (s + e)>>1;
    int lsame = mid - s + 1;
    int i,same = 0;
    for(i = s;i <= e;i ++)
        if(num[dp][i] < lcm[mid])
            lsame --;
    int lpos = s,rpos = mid + 1;
    for(i = s;i <= e;i ++)
    {
        if(i == s)
            lefta[dp][i] = 0;
        else
            lefta[dp][i] = lefta[dp][i - 1];
        if(num[dp][i] < lcm[mid])
        {
            lefta[dp][i] ++;
            num[dp + 1][lpos ++] = num[dp][i];
        }
        else
        {
            if(num[dp][i] > lcm[mid])
                num[dp + 1][rpos ++] = num[dp][i];
            else
            {
                if(same < lsame)
                {
                    same ++;
                    lefta[dp][i] ++;
                    num[dp + 1][lpos ++] = num[dp][i];
                }
                else
                    num[dp + 1][rpos ++] = num[dp][i];
            }
        }
    }
    build(s,mid,nm<<1,dp + 1);
    build(mid + 1,e,nm<<1|1,dp + 1);
}
int query(int s,int e,int nm,int dp,int l,int r,int k)
{
    if(s == e)
        return num[dp][s];
    int mid = (s + e)>>1;
    int sa,sb;
    if(s == l)
        sb = 0;
    else
        sb = lefta[dp][l - 1];
    sa = lefta[dp][r] - sb;
    if(sa >= k)
    {
        int newl = s + sb;
        int newr = s + sb + sa - 1;
        return query(s,mid,nm<<1,dp + 1,newl,newr,k);
    }
    else
    {
        int bb = l - s - sb;
        int b = r - l + 1 - sa;
        int newl = mid + bb + 1;
        int newr = mid + bb + b;
        return query(mid + 1,e,nm<<1|1,dp + 1,newl,newr,k - sa);
    }
}
ll getans1(int a,int b,int c)
{
    int mid,la,ra;
    la = a;ra = b;
    while(la <= ra)
    {
        mid = (la + ra)>>1;
        int tmp = query(1,n,1,0,a,b,mid);
        if(tmp == c)
            return mid - a + 1;
        else if(tmp > c)
                ra = mid - 1;
            else
                la = mid + 1;
    }
}
int main()
{
    int i;
    int t,q,a,b,c,cas = 0;
    while(~scanf("%d",&q))
    {
        n = m = 1;
        while(q --)
        {
            scanf("%s",str);
            if(str[0] == 'I')
            {
                scanf("%d",&lcm[n]);
                num[0][n] = lcm[n ++];
            }
            else
            {
                if(str[strlen(str) - 1] == '1')
                {
                    scanf("%d%d%d",&a,&b,&c);
                    op[m].optype = 0;
                    op[m].a = a;op[m].b = b;op[m ++].c = c;
                }
                else if(str[strlen(str) - 1] == '2')
                    {
                        scanf("%d",&a);
                        op[m].optype = 1;
                        op[m].a = 1;op[m].b = n - 1;
                        op[m ++].c = a;
                    }
                    else
                    {
                        scanf("%d",&a);
                        op[m].optype = 2;
                        op[m].a = 1;op[m].b = n - 1;
                        op[m ++].c = a;
                    }
            }
        }
        sort(lcm + 1,lcm + n);
        n --;
        build(1,n,1,0);
        ans[1] = ans[0] = ans[2] = 0;
        for(i = 1;i < m;i ++)
        {
            if(op[i].optype == 1)
            {
                int tmp = getans1(op[i].a,op[i].b,op[i].c);
                ans[op[i].optype] += tmp;
            }
            else
            {
                int tmp = query(1,n,1,0,op[i].a,op[i].b,op[i].c);
                ans[op[i].optype] += tmp;
            }
        }
        printf("Case %d:\n%I64d\n%I64d\n%I64d\n",++cas,ans[0],ans[1],ans[2]);
    }
    return 0;
}
//265MS	15908K


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章