题解:
没想到线段树可以做这种题。。。这道题通过枚举留最高的桌子,首先比这个高的桌子肯定全没了,剩下的如果这个桌子的数量已经大于2倍那就不用再砍脚了,如果不够那么就砍掉前k小权值的桌脚。所以线段树维护的权值L到R区间内的数量和权值和。这道题也可以用权值线段树来做,之后来补上来。
#include <bits/stdc++.h>
#define int long long
using namespace std;
#define lson u<<1
#define rson u<<1|1
const int N=1e5+10;
int a[N],b[N],cnt[N],sum[N];
vector<int> g[N];
struct Node
{
int l,r;
int num,sum;
}tr[N<<2];
void pushup(int u)
{
tr[u].num=tr[lson].num+tr[rson].num;
tr[u].sum=tr[lson].sum+tr[rson].sum;
}
void build(int u,int l,int r)
{
tr[u]={l,r,0,0};
if(l==r) return ;
int mid=l+r>>1;
build(lson,l,mid),build(rson,mid+1,r);
}
int query(int u,int k)
{
if(tr[u].l==tr[u].r){
return tr[u].l*k;
}
if(tr[lson].num==k) return tr[lson].sum;
else if(tr[lson].num>=k) return query(lson,k);
else return tr[lson].sum+query(rson,k-tr[lson].num);
}
void modify(int u,int x)
{
if(tr[u].l==tr[u].r){
tr[u].num++;
tr[u].sum+=x;
return;
}
int mid=tr[u].l+tr[u].r>>1;
if(x<=mid) modify(lson,x);
else modify(rson,x);
pushup(u);
}
signed main()
{
int n,maxn=-1,maxa=-1; scanf("%lld",&n);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]),cnt[a[i]]++,maxa=max(maxa,a[i]);
for(int i=1;i<=n;i++){
scanf("%lld",&b[i]);
sum[a[i]]+=b[i];
maxn=max(maxn,b[i]);
g[a[i]].push_back(b[i]);
}
for(int i=1;i<=maxa;i++){
cnt[i]+=cnt[i-1];
sum[i]+=sum[i-1];
}
build(1,1,maxn);
int ans=1e18;
for(int i=1;i<=maxa;i++){
int tmp=sum[maxa]-sum[i];
if(g[i].size()*2>cnt[i]){
ans=min(ans,tmp);
}else{
int k=cnt[i]-g[i].size()*2+1;
ans=min(ans,tmp+query(1,k));
}
for(int j=0;j<g[i].size();j++) modify(1,g[i][j]);
}
printf("%lld\n",ans);
}