F - Disharmony Trees HDU - 3015 树状数组

传送门

题意：广场上有n棵树。两棵树之间的不和谐值与两种称为远和短的值有关。

例如，如果有5棵树，X坐标是3,3,1,3,4.。它们的D可能是2 2 1 2 5。将两棵X坐标为D1和D2的树的距离定义为F = abs(D1-D2)。短的定义与远的相似。

例如，如果有5棵树的高度是4 1 9 7 4。它们的H可能是2 1 5 4 2。高度为H1和H2的两棵树的短树定义为S=min(H1,H2)。二树的不和谐值定义为F*S。

请告诉Sophia每两棵树之间不和谐值的总和。

思路：离散化，根据排序后的值进行计算

但是这题数据过大，正常暴力是不可行的

首先将树的H进行从大到小排序，此时H是按顺序的，就不需要在比较H大小判断用哪个了

然后计算从1到i-1所有数与i的F（总），再乘i的H。

现在研究1到i-1这些数据：由于F取得是绝对值，所以要分两种情况：比x大，比x小

F（总）=F（小）+F(大）=num*x-ans+（ans（总）-ans）-（i-num-1）*x；

我们需要用两个数组分别标记比x小的数的个数num，比x小的数的总和ans

但是怎么求每一个点的num值和ans值？这里其实就是逆序数操作了

代码：

#include<stdio.h>
#include<algorithm>
#include<math.h>
using namespace std;
#include<string.h>
#include<iostream>
#define ll long long
struct note
{
    ll x,y;
} k[100100];
int cmp(note k,note p)
{
    if(k.y==p.y) return k.x<p.x;
    return k.y>p.y;
}
ll a[100100],b[100100],num[100100],ans[100100];
ll n;
ll lowbit(ll x)
{
    return x&(-x);
}
void add(ll x,ll v,ll val[])
{
    for(ll i=x; i<=n; i+=lowbit(i))
    {
        val[i]+=v;
    }
}
ll getsum(ll x,ll val[])
{
    ll ans=0;
    for(ll i=x; i; i-=lowbit(i))
    {
        ans+=val[i];
    }
    return ans;
}
int main()
{
    while(~scanf("%lld",&n))
    {
        memset(num,0,sizeof(num));
        memset(ans,0,sizeof(ans));
        for(ll i=1; i<=n; i++)
        {
            scanf("%lld %lld",&k[i].x,&k[i].y);
            a[i]=k[i].x;
            b[i]=k[i].y;
        }
        sort(a+1,a+n+1);
        sort(b+1,b+1+n);
        for(ll i=1; i<=n; i++)
        {
            k[i].x=lower_bound(a+1,a+n+1,k[i].x)-a;
            k[i].y=lower_bound(b+1,b+n+1,k[i].y)-b;
        }
        sort(k+1,k+n+1,cmp);
        ll sum=0;
        for(ll i=1; i<=n; i++)
        {
            ll ha=getsum(n,ans);  //查找n之前的数字的总和
            ll w=getsum(k[i].x,num);
            ll v=getsum(k[i].x,ans);
            sum+=(w*k[i].x-v+(ha-v)-(i-w-1)*k[i].x)*k[i].y;
            add(k[i].x,1,num);
            add(k[i].x,k[i].x,ans);
        }
        printf("%lld\n",sum);
    }
}