HDU - 3015 Disharmony Trees （树状数组）

题目链接

题意：

在路旁长了一些树，但是看起来不是很和谐，不和谐的因素有 $FAR$ 和 $SHORT$，不和谐的值 $disharmony= F * S$;

$FAR$        定义为 $F =$ 两棵树之间 $x$ 坐标 大小排名 之差的绝对值

$SHORT$  定义为 $S =$ 两个树之间高度最低的排名

求所有树的不和谐值的总和。

思路：

和$MooFest$的思路很类似，因为要求 坐标 大小排名 之差的绝对值 ，所以我们要同时记录个数和值；

先将所有的树按照高度由高到低排序，那么我们每一次所用的S，就是当前树高度的排名；

/*
* @Author: windystreet
* @Date:   2018-08-09 10:44:50
* @Last Modified by:   windystreet
* @Last Modified time: 2018-08-09 15:56:07
*/
#include<bits/stdc++.h>

using namespace std;

#define X first
#define Y second
#define eps  1e-5
#define gcd __gcd
#define pb push_back
#define PI acos(-1.0)
#define lowbit(x) (x)&(-x)
#define bug printf("!!!!!\n");
#define mem(x,y) memset(x,y,sizeof(x))

typedef long long LL;
typedef long double LD;
typedef pair<int,int> pii;
typedef unsigned long long uLL;

const int maxn = 1e5+20;
const int INF  = 1<<30;
const int mod  = 1e9+7;

struct node
{
    int  x,h;
}s[maxn];

int xx[maxn],hh[maxn];

LL tree[maxn][2]; // 0 num
int n;            // 1 xsum
bool cmp(node a, node b){
    return a.h>b.h||(a.h==b.h&&a.x>b.x);
}

void init(){
    for(int i=0;i<=n+10;i++){
        for(int j=0;j<2;j++){
            tree[i][j] = 0;
        }
    }
}

void add(int x,int v,int flag){
    for(int i=x;i<=n;i+=lowbit(i)){
        tree[i][flag] += v;
    }return;
}

LL query(int x,int flag){
    LL res = 0;
    for(int i=x;i;i-=lowbit(i)){
        res += tree[i][flag];
    }
    return res; 
}

void solve(){
    while(scanf("%d",&n)!=EOF){
        init();
        for(int i=1;i<=n;i++){
            scanf("%d%d",&s[i].x,&s[i].h);
            xx[i] = s[i].x; hh[i] = s[i].h;
        }
        sort(xx+1,xx+1+n);sort(hh+1,hh+1+n);sort(s+1,s+1+n,cmp);
        LL P,S,ans = 0;
        for(int i=1;i<=n;i++){
            int pos = lower_bound(xx+1,xx+1+n,s[i].x)-xx;        // 获取当前树的坐标排名
            int hi  = lower_bound(hh+1,hh+1+n,s[i].h)-hh;        // 获取当前树的高度排名
            P = pos*query(pos,0) - query(pos,1) + query(n,1)-query(pos,1)-(query(n,0)-query(pos,0))*pos;
            // 以上是计算之前已经存在的树与当前树的排名差的绝对值，分别是计算当前树位置之间的 和 之后的
            S = hi;                                                
            ans += P*S;                                          // 更新答案
            add(pos,1,0);add(pos,pos,1);                        
        }
        printf("%lld\n", ans);
    }
    
    return;
}

int main()
{
//    freopen("in.txt","r",stdin);
//    freopen("out.txt","w",stdout);
//    ios::sync_with_stdio(false);
    int t = 1;
    //scanf("%d",&t);
    while(t--){
    //    printf("Case %d: ",cas++);
        solve();
    }
    return 0;
}