POJ 1990

1. Brute Force方法导致TLE

2. 数组树做法,先按照v升序排序。然后对排序好的Cow循环,假设在i位置的cow,那么0到i-1位置的cow的v都比当前v小,则可以利用公式

v * (prevCnt * x - prevDist + totalDist - prevDist - (i - prevCnt) * x)算出。

v * (prevCnt * x - prevDist)为左半部分

v * (totalDist - prevDist - (i - prevCnt) * x)为右半部分


资料:关于lowbit函数

http://wenku.baidu.com/view/3f9579c689eb172ded63b7c4.html

#include <iostream>
#include <algorithm>
using namespace std;


const int MAXN = 20001;


int cnt[MAXN];
int dist[MAXN];


struct Cow
{
int x;
int v;
bool operator<(const Cow& c) const
{
return v < c.v;
};
}cows[20001];


int lowbit(int x)
{
return x & (-x);
};


void insert(int *a, int p, int value)
{
while(p <= MAXN)
{
a[p] += value;
p += lowbit(p);
}


};


int sum(int* a, int p)
{
int res = 0;
while(p > 0)
{
res += a[p];
p -= lowbit(p);
}
return res;
};






int main()
{
int n;
cin >> n;
for(int i = 0; i < n; i++)
{
cin >> cows[i].v >> cows[i].x;
}
sort(cows, cows + n);


long long res = 0;
int totalDist = 0;

for(int i = 0; i < n; i++)
{
int x = cows[i].x;
int v = cows[i].v;
int prevDist = sum(dist, x);
int prevCnt = sum(cnt, x);
res += (long long)v * (prevCnt * x - prevDist + totalDist - prevDist - (i - prevCnt) * x);
insert(dist, x, x);
insert(cnt, x, 1);
totalDist += x;
}
cout << res;
return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值