程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> HDU 3473 Minimum Sum (劃分樹)

HDU 3473 Minimum Sum (劃分樹)

編輯:C++入門知識

題意:給定一個數組,有Q次的詢問,每次詢問的格式為(l,r),表示求區間中一個數x,使得sum = sigma|x - xi|最小(i在[l,r]之間),輸出最小的sum。   思路:本題一定是要O(nlogn)或更低復雜度的算法。首先很容易得出這個x的值一定是區間(l,r)的中位數的取值,排序之後,也就是假設區間(l,r)長度為len ,則中位數就是該區間的第(r - l) / 2 - 1小的元素,求一個區間的第K小元素的算法很自然地會想到劃分樹, 而且劃分樹的查詢復雜度為:O(logn),正好可以解決此題。   算法確定了之後就是具體的實現過程了,普通的劃分樹求的是區間內的第k小的元素,而這題是要求差值,也就是說我們不但要求出區間的第k小的元素,還要求出所有比中位數小的數 lsum,當然比中位數大的數的和可以根據區間的數的總和和lsum求得,因此不需要額外求。這樣我們只需要在劃分樹建樹的時候增加一個lsum[ ][i] 數組就可以了, 這個數組保存的是,在step層,在i前面被劃分到左子樹的元素之和。這樣我們就可以求出最後的解了。  

#include <iostream>  
#include <algorithm>  
#include <cmath>  
#include<functional>  
#include <cstdio>  
#include <cstdlib>  
#include <cstring>  
#include <string>  
#include <vector>  
#include <set>  
#include <queue>  
#include <stack>  
#include <climits>//形如INT_MAX一類的  
#define MAX 100005  
#define INF 0x7FFFFFFF  
#define L(x) x << 1  
#define R(x) x << 1 | 1  
using namespace std;  
  
struct Seg_Tree {  
    int l,r,mid;  
} tr[MAX*4];  
int sorted[MAX];  
int lef[20][MAX];  
int val[20][MAX];  
__int64 lsum[20][MAX];  
__int64 sum[MAX];  
__int64 summ;  
  
void build(int l,int r,int step,int x) {  
    tr[x].l = l;  
    tr[x].r = r;  
    tr[x].mid = (l + r) >> 1;  
    if(tr[x].l == tr[x].r) return ;  
    int mid = tr[x].mid;  
    int lsame = mid - l + 1;//lsame表示和val_mid相等且分到左邊的  
    for(int i = l ; i <= r ; i ++) {  
        if(val[step][i] < sorted[mid]) {  
            lsame --;//先假設左邊的數(mid - l + 1)個都等於val_mid,然後把實際上小於val_mid的減去  
        }  
    }  
    int lpos = l;  
    int rpos = mid + 1;  
    int same = 0;  
    for(int i = l ; i <= r ; i ++) {  
        if(i == l) {  
            lef[step][i] = 0;//lef[i]表示[ tr[x].l , i ]區域裡有多少個數分到左邊  
            lsum[step][i] = 0;  
        } else {  
            lef[step][i] = lef[step][i-1];  
            lsum[step][i] = lsum[step][i-1];  
        }  
        if(val[step][i] < sorted[mid]) {  
            lef[step][i] ++;  
            lsum[step][i] += val[step][i];  
            val[step + 1][lpos++] = val[step][i];  
        } else if(val[step][i] > sorted[mid]) {  
            val[step+1][rpos++] = val[step][i];  
        } else {  
            if(same < lsame) {//有lsame的數是分到左邊的  
                same ++;  
                lef[step][i] ++;  
                lsum[step][i] += val[step][i];  
                val[step+1][lpos++] = val[step][i];  
            } else {  
                val[step+1][rpos++] = val[step][i];  
            }  
        }  
    }  
    build(l,mid,step+1,L(x));  
    build(mid+1,r,step+1,R(x));  
}  
  
int query(int l,int r,int k,int step,int x) {  
    if(l == r) {  
        return val[step][l];  
    }  
    int s;//s表示[l , r]有多少個分到左邊  
    int ss;//ss表示 [tr[x].l , l-1 ]有多少個分到左邊  
    __int64 tmp = 0;  
    if(l == tr[x].l) {  
        tmp = lsum[step][r];  
        s = lef[step][r];  
        ss = 0;  
    } else {  
        tmp = lsum[step][r] - lsum[step][l-1];  
        s = lef[step][r] - lef[step][l-1];  
        ss = lef[step][l-1];  
    }  
    if(s >= k) {//有多於k個分到左邊,顯然去左兒子區間找第k個  
        int newl = tr[x].l + ss;  
        int newr = tr[x].l + ss + s - 1;//計算出新的映射區間  
        return query(newl,newr,k,step+1,L(x));  
    } else {  
        summ += tmp;  
        int mid = tr[x].mid;  
        int bb = l - tr[x].l - ss;//bb表示 [tr[x].l , l-1 ]有多少個分到右邊  
        int b = r - l + 1 - s;//b表示 [l , r]有多少個分到右邊  
        int newl = mid + bb + 1;  
        int newr = mid + bb + b;  
        return query(newl,newr,k-s,step+1,R(x));  
    }  
}  
  
void solve(int l,int r) {  
    l ++; r ++;  
    summ = 0;  
    int k = (r - l) / 2 + 1;  
    __int64 mid = query(l,r,k,0,1);  
    __int64 ans = (k - 1) * mid - summ;  
    ans += sum[r] - sum[l-1] - summ - (r - l - k + 2) * mid;  
    printf("%I64d\n",ans);  
}  
  
int n,m,l,r;  
int main() {  
    int T;  
    cin >> T;  
    int ca = 1;  
    while(T--) {  
        scanf("%d",&n);  
        memset(sum,0,sizeof(sum));  
        for(int i=1; i<=n; i++) {  
            scanf("%d",&val[0][i]);  
            sorted[i] = val[0][i];  
            sum[i] = sum[i-1] + val[0][i];  
        }  
        sort(sorted+1,sorted+1+n);  
        build(1,n,0,1);  
        scanf("%d",&m);  
        printf("Case #%d:\n",ca++);  
        for(int i=0; i<m; i++) {  
            scanf("%d%d",&l,&r);  
            solve(l,r);  
        }  
        puts("");  
    }  
}  

 

 

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved