HDU 3473 划分树

点击打开链接

题意:问你给定区间的给出的表达式的最小值

思路:看了网上题解,都说中位数便是最优的解,证明不会,那么跟着思路做就行了,用划分树找出中位数然后在划分树中加入区间的前缀和即可,加的时候只要查询的值走到右子树那么就加起来,因为它向右走了,那么左边全是比它小的要加起来,小于它的个数一样统计一下即可,然后有了这两个推个小公式就行了

#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef __int64 ll;
const int maxn=100010;
int t[20][maxn],sum[20][maxn],as[maxn];
ll cnt[maxn],Lcnt[20][maxn],Lcnt1,Lsum;
void buildtree(int le,int ri,int p){
    int mid=(le+ri)>>1;
    int lm=mid-le+1,ls=le,rs=mid+1;
    for(int i=le;i<=ri;i++){
        if(t[p][i]>1;
    if(le==ri) return t[node][le];
    int s,ss;
    ll tmp=0;
    if(l==le) s=0,ss=sum[node][r],tmp=Lcnt[node][r];
    else s=sum[node][l-1],ss=sum[node][r]-s,tmp=Lcnt[node][r]-Lcnt[node][l-1];
    if(k<=ss) return query(le+s,le+sum[node][r]-1,k,le,mid,node+1);
    else{//去右区间查询,左边全是小的加起来
        Lcnt1+=ss;Lsum+=tmp;
        return query(mid+1-le+l-s,mid+1-le+r-sum[node][r],k-ss,mid+1,ri,node+1);
    }
}
int main(){
    int n,m,a,b,T,cas=1;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);cnt[0]=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&as[i]);
            t[0][i]=as[i];
            cnt[i]=cnt[i-1]+as[i];
        }
        sort(as+1,as+1+n);
        buildtree(1,n,0);
        scanf("%d",&m);
        printf("Case #%d:\n",cas++);
        while(m--){
            scanf("%d%d",&a,&b);
            a++;b++;
            Lcnt1=0;Lsum=0;
            ll k=query(a,b,(b-a+2)/2,1,n,0);
            ll rcnt=b-a-Lcnt1;
            ll rsum=cnt[b]-cnt[a-1]-k-Lsum;
            ll tt1=k*Lcnt1-Lsum;
            ll tt2=rsum-k*rcnt;
            ll ans=tt1+tt2;//那个小公式很好推
            printf("%I64d\n",ans);
        }
        printf("\n");
    }
    return 0;
}

你可能感兴趣的:(线段树&树状数组,数据结构,线段树)