题意:给一个长度为n的数列a,q组询问,每组询问求将li到ri的区间中的数哈夫曼编码的长度。
莫队,对于每一个询问处理该区间内出现次数为x的字符个数。
对于次数小于
n√
的字符,从小到大枚举出现次数,对于一种出现次数x两两配对,在
x∗2
插入x的个数/2。
对于出现次数大于
n√
的字符,用构建哈夫曼树的方法处理。
时间复杂度 O(nn√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;
}