【Codeforces700D】Huffman Coding on Segment

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

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

#include <bits/stdc++.h>
#define gc getchar()
#define ll long long
#define N 100009
using 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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值