题意:广场上有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);
}
}