题意:
给你n个数,问区间
我撕烤了大概很久线段树怎么维护。然后一看题解告诉我是莫队!(wtf??)
按出现次数分为大于n√和小于等于n√的。小于等于的从小到大无脑合并,如果合出了大于n√的就和原来的大于n√的一起加入一个优先队列,最后处理完后挑小的两个合并就好啦。
时间复杂度O(nn√logn),可以过。
#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;
}