【Codeforces700D】Huffman Coding on Segment

来源:互联网 发布:php企业网站源码 编辑:程序博客网 时间:2024/05/29 09:05

题意:
给你n个数,问区间[l,r]中的数用二进制哈夫曼编码编码后的最短长度。

我撕烤了大概很久线段树怎么维护。然后一看题解告诉我是莫队!(wtf??)
按出现次数分为大于n和小于等于n的。小于等于的从小到大无脑合并,如果合出了大于n的就和原来的大于n的一起加入一个优先队列,最后处理完后挑小的两个合并就好啦。
时间复杂度O(nnlogn),可以过。

#include <bits/stdc++.h>#define gc getchar()#define ll long long#define N 100009using namespace std;int n,a[N],Q,limit,block[N],num[N],number[N],Ans[N],Number[N];vector<int> Big;struct qry{    int l,r,pos;    bool operator <(const qry &rhs)    {        return block[l]<block[rhs.l]||(block[l]==block[rhs.l]&&r<rhs.r);    }}q[N];int read(){    int x=1;    char ch;    while (ch=gc,ch<'0'||ch>'9') if (ch=='-') x=-1;    int s=ch-'0';    while (ch=gc,ch>='0'&&ch<='9') s=s*10+ch-'0';    return x*s;}void add(int x,int y){    number[num[x]]--;    num[x]+=y;    number[num[x]]++;}int calc(){    priority_queue<int,vector<int>,greater<int> > qq;    for (int i=0;i<(int)Big.size();i++)        if (num[Big[i]]>limit) qq.push(num[Big[i]]);    for (int i=1;i<=limit;i++) Number[i]=number[i];    int ret=0,last=0;    for (int i=1;i<=limit;i++)        if (Number[i])        {            if (last)            {                Number[i]--,ret+=last+i;                if (last+i<=limit) Number[last+i]++;                else qq.push(last+i);                last=0;            }            if (Number[i]&1)                Number[i]--,last=i;            ret+=Number[i]*i;            if (2*i<=limit) Number[2*i]+=Number[i]/2;            else                for (int j=1;j<=Number[i]/2;j++) qq.push(i*2);        }    if (last) qq.push(last);    while ((int)qq.size()>1)    {        int x1=qq.top();        qq.pop();        int x2=qq.top();        qq.pop();        ret+=x1+x2,qq.push(x1+x2);    }    return ret;}int main(){    n=read();    limit=sqrt(n);    for (int i=1;i<=n;i++)    {        a[i]=read();        if ((++num[a[i]])==limit) Big.push_back(a[i]);        if (num[a[i]]==1) number[0]++;     }    memset(num,0,sizeof(num));    Q=read();    for (int i=1;i<=Q;i++)        q[i].l=read(),q[i].r=read(),q[i].pos=i;    for (int i=1;i<=n;i++)        block[i]=i/limit+1;    sort(q+1,q+Q+1);    int l=1,r=0;    for (int i=1;i<=Q;i++)    {        while (r<q[i].r) add(a[++r],1);        while (l>q[i].l) add(a[--l],1);        while (r>q[i].r) add(a[r--],-1);        while (l<q[i].l) add(a[l++],-1);        Ans[q[i].pos]=calc();     }    for (int i=1;i<=Q;i++)        printf("%d\n",Ans[i]);    return 0;}
原创粉丝点击