[ 莫队 哈夫曼树 ] Codeforces700D Huffman Coding on Segment

莫队。每次询问不断将值小于 $\sqrt n$ 的合并，剩下的用构建哈夫曼树的方法来做就可以了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100010;
struct Node{
  int l,r,id;
}c[N];
int k,n,m,S,L=1,R;
int a[N],b[N];
int cnt[N],s[N],sum[N];
bool v[N];
ll Ans[N];
vector<int>g;
inline bool Cmp(Node x,Node y){
  return b[x.l]<b[y.l]||(b[x.l]==b[y.l]&&x.r<y.r);
}
inline void Update(int x,int y){
  sum[cnt[x]]--;
  cnt[x]+=y;
  sum[cnt[x]]++;
}
inline ll Calc(){
  ll Ans=0;
  priority_queue<int,vector<int>,greater<int> >Q;
  for(int i=0;i<g.size();i++)
    if(cnt[g[i]]>S)Q.push(cnt[g[i]]);
  int lst=0;
  for(int i=1;i<=S;i++)s[i]=sum[i];
  for(int i=1;i<=S;i++)
    if(s[i]){
      if(lst){
                Ans+=i+lst;
                if(i+lst<=S)s[i+lst]++;else Q.push(i+lst);
                s[i]--;lst=0;
      }
      if(s[i]&1)lst=i;
      Ans+=1ll*(s[i]>>1)*(i<<1);
      if((i<<1)<=S)s[i<<1]+=s[i]>>1;else
                for(int j=1;j<=(s[i]>>1);j++)Q.push(i<<1);
    }
    if(lst)Q.push(lst);
  while(Q.size()>1){
    int x=Q.top();Q.pop();
    x+=Q.top();Q.pop();
    Ans+=x;Q.push(x);
  }
  return Ans;
}
int main(){
  scanf("%d",&n);
  for(int i=1;i<=n;i++)scanf("%d",&a[i]),cnt[a[i]]++;
  scanf("%d",&m);
  for(int i=1;i<=m;i++)scanf("%d%d",&c[i].l,&c[i].r),c[i].id=i;
  S=sqrt(n);
  for(int i=1;i<=n;i++)
    if(cnt[a[i]]>S&&!v[a[i]])
      g.push_back(a[i]),v[a[i]]=1;
  memset(cnt,0,sizeof(cnt));
  for(int i=1;i<=n;i++)b[i]=(i-1)/S+1;
  sort(c+1,c+m+1,Cmp);
  for(int i=1;i<=m;i++){
    while(R<c[i].r)Update(a[++R],1);
    while(R>c[i].r)Update(a[R--],-1);
    while(L>c[i].l)Update(a[--L],1);
    while(L<c[i].l)Update(a[L++],-1);
    Ans[c[i].id]=Calc();
  }
  for(int i=1;i<=m;i++)printf("%I64d\n",Ans[i]);
  return 0;
}