codeforces 700D 哈夫曼树 莫队

题意：给一个长度为n的数列a，q组询问，每组询问求将li到ri的区间中的数哈夫曼编码的长度。

莫队，对于每一个询问处理该区间内出现次数为x的字符个数。
对于次数小于$\sqrt{n}$ 的字符，从小到大枚举出现次数，对于一种出现次数x两两配对，在$x*2$ 插入x的个数/2。
对于出现次数大于$\sqrt{n}$ 的字符，用构建哈夫曼树的方法处理。

时间复杂度$O(n\sqrt{n}logn)$

能过。。。。

#include <bits/stdc++.h>
using namespace std;
#define N 110000
int  n,q,size;
vector<int>st;
int a[N],cnt[N],num[N],c[N],ans[N],num1[N];
struct node
{
    int l,r,bel;
    friend bool operator < (const node &r1,const node &r2)
    {
        if(r1.l/size==r2.l/size)return r1.r<r2.r;
        return r1.l<r2.l;
    };
}b[N];
void update(int x,int v)
{
    num[c[x]]--;
    c[x]+=v;
    num[c[x]]++;
}
int cal()
{
    int pre=0,ret=0;
    priority_queue<int,vector<int>,greater<int> >que;
    for(int i=0;i<st.size();i++)
        if(c[st[i]]>size)que.push(c[st[i]]);
    for(int i=1;i<=size;i++)num1[i]=num[i];
    for(int i=1;i<=size;i++)
        if(num1[i])
        {
            if(pre)
            {
                num1[i]--;
                if(pre+i>size)que.push(pre+i);
                else num1[pre+i]++;
                ret+=pre+i;pre=0;
            }
            if(num1[i]&1)
                num1[i]--,pre=i;
            ret+=num1[i]*i;
            if(i*2>size)
                for(int j=1;j<=num1[i]/2;j++)que.push(i*2);
            else num1[i*2]+=num1[i]/2;
        }
    if(pre)que.push(pre);
    while(que.size()>1)
    {
        int r1=que.top();que.pop();
        int r2=que.top();que.pop();
        que.push(r1+r2);ret+=r1+r2;
    }
    return ret;
}
int main()
{
    //freopen("tt.in","r",stdin);
    scanf("%d",&n);
    size=sqrt(n)+1;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        cnt[a[i]]++;
        if(cnt[a[i]]==size)
            st.push_back(a[i]);
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++)
    {
        scanf("%d%d",&b[i].l,&b[i].r);
        b[i].bel=i;
    }
    sort(b+1,b+1+q);
    int r=0,l=1;
    for(int i=1;i<=q;i++)
    {
        while(l<b[i].l)update(a[l],-1),l++;
        while(l>b[i].l)l--,update(a[l],1);
        while(r<b[i].r)r++,update(a[r],1);
        while(r>b[i].r)update(a[r],-1),r--;
        ans[b[i].bel]=cal();
    }
    for(int i=1;i<=q;i++)
        printf("%d\n",ans[i]);
    return 0;
}