HDU 3473 Minimum Sum(划分树)

题意:
给出一个长度为n的序列,Q次询问,每次询问给出l,r让你求出下列式子的最小值
∑ i = l r ∣ x − x i ∣ \sum_{i=l}^{r}|x-x_i| i=lrxxi


题解:
很容易知道这道题无非就是想每次询问都找到中位数,然后求一下和。

中位数我们用划分树可以解决,求和的话我们可以再加一个数组sum[i][j]表示第i层的[L,i]之间的数中分到左子树的所有数字之和,有了这个右儿子的和就直接用[L,R]减去左儿子的和就好了

然后将式子化简以后可以得到:

s u m r − s u m l + ( l n u m − r n u m ) ∗ m i d sum_r-sum_l+(l_{num}-r_{num})*mid sumrsuml+(lnumrnum)mid

l n u m l_{num} lnum表示比中位数小的个数, r n u m r_{num} rnum表示比中位数大的数的个数


AC代码:

#pragma GCC optimize(2)
#include<bits/stdc++.h>
#include<ext/rope>
using namespace std;
using namespace __gnu_cxx;
typedef long long LL;
const int MAXN = 1e5+10;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
int val[20][MAXN],num[20][MAXN],sorted[MAXN]; LL sum[20][MAXN],res;
inline void build(int l,int r,int dep){
    if(l==r){ sum[dep][l]=sum[dep][l-1]+val[dep][l];return; }
    int mid=(l+r)>>1,same=mid-l+1;
    for(int i=l;i<=r;i++){
        if(val[dep][i]<sorted[mid]) same--;
        sum[dep][i]+=sum[dep][i-1]+val[dep][i];
    }
    int ln=l,rn=mid+1;
    for(int i=l;i<=r;i++){
        if((val[dep][i]<sorted[mid]) || (val[dep][i]==sorted[mid]&&same>0)){
            val[dep+1][ln++]=val[dep][i];
            if(val[dep][i]==sorted[mid]) same--;
        }else val[dep+1][rn++]=val[dep][i];
        num[dep][i]=num[dep][l-1]+ln-l;
    }
    build(l,mid,dep+1); build(mid+1,r,dep+1);
}
inline int query(int dep,int l,int r,int ql,int qr,int k){
    if(ql==qr) return val[dep][ql];
    int mid=(l+r)>>1,tot=num[dep][qr]-num[dep][ql-1];
    if(tot>=k){
        int s = qr-l+1-(num[dep][qr]-num[dep][l-1])+mid;
        int t = ql-l-(num[dep][ql-1]-num[dep][l-1])+mid;
        res += sum[dep+1][s]-sum[dep+1][t];
        int ll=l+num[dep][ql-1]-num[dep][l-1];
        int rr=ll+tot-1;
        return query(dep+1,l,mid,ll,rr,k);
    }else{
        int s = l+num[dep][ql-1]-num[dep][l-1];
        int t = s+tot-1;
        res -= sum[dep+1][t]-sum[dep+1][s-1];
        int rr=qr+num[dep][r]-num[dep][qr];
        int ll=rr-(qr-ql+1-tot)+1;
        return query(dep+1,mid+1,r,ll,rr,k-tot);
    }
}
signed main(){
#ifndef ONLINE_JUDGE
    freopen("C:\\Users\\Administrator\\Desktop\\in.txt","r",stdin);
#endif // ONLINE_JUDGE
    int T,cas=0; scanf("%d",&T);
    while(T--){
        int n; scanf("%d",&n);
        memset(sum,0,sizeof(sum)); memset(val,0,sizeof(val));
        for(int i=1;i<=n;i++) scanf("%d",&val[0][i]),sorted[i]=val[0][i];
        sort(sorted+1,sorted+n+1);
        build(1,n,0);
        int Q; scanf("%d",&Q);
        printf("Case #%d:\n",++cas);
        while(Q--){
            int l,r; scanf("%d%d",&l,&r); l++;r++;
            res=0;
            int ans = query(0,1,n,l,r,(r-l+2)/2);
            if((r-l+1)%2==0) res-=ans;
            printf("%lld\n",res);
        }
        puts("");
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值