首页 > 代码库 > POJ 1990 MooFest 树状数组
POJ 1990 MooFest 树状数组
题目大意:Farmer John又来恶心我们了!这次他带来了一些牛,这些牛排成一列,他们的位置给出,每一个牛有一个音调。这些牛每两只牛之间都要互相交流,但是交流的时候会有一些花费,i,j两只牛的cost = max(vi,vj) * |posi - posj|。求所有牛之间互相交流的cost和。
思路:一开始我还以为是最大或者最小花费,后来仔细读题发现想多了,就是单纯的统计,但是数据范围2w显然不能n^2的统计,我们需要来想一些优化的方法。
对于所有的牛对来说,最后计算cost的时候是决定于音调高的那一头牛身上。所以就考虑能不能把音调从低到高排序,然后不断的往其中加牛。这样每次的牛加进去的时候,之前的所有牛的声调都比加进去的低,也就可以用新加进去的牛来计算了。
注意到每一个新加进来的牛需要向ans中累加Σvx * |posx - posi|这么多,vx已经知道了,那么怎么求距离之和呢?我的方法是维护4个树状数组,分别统计在x前面的所有点到原点的距离之和,x前面有多少牛;x后面的点的到原点的距离之和,x后面有多少牛。这样就可以通过cnt * posx - Σpos_pred来计算出x前面的距离x之和,后面的也同理。
分析下时间复杂度,O(nlogn),水过吧。。
CODE:
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define MAX 200010 using namespace std; struct Complex{ int val,pos; bool operator <(const Complex &a)const { return val < a.val; } void Read() { scanf("%d%d",&val,&pos); } }point[MAX]; int points; long long pred[MAX],cnt1[MAX]; long long succ[MAX],cnt2[MAX]; inline void FixPred(int x); inline void FixSucc(int x); inline long long GetPred(int x); inline long long GetSucc(int x); int main() { cin >> points; for(int i = 1;i <= points; ++i) point[i].Read(); sort(point + 1,point + points + 1); long long ans = 0; for(int i = 1;i <= points; ++i) { ans += GetPred(point[i].pos) * point[i].val; ans += GetSucc(point[i].pos) * point[i].val; FixPred(point[i].pos); FixSucc(point[i].pos); } cout << ans << endl; return 0; } inline void FixPred(int x) { for(int i = x;i < MAX;i += i&-i) { pred[i] += x; ++cnt1[i]; } } inline void FixSucc(int x) { for(int i = x;i;i -= i&-i) { succ[i] += x; ++cnt2[i]; } } inline long long GetPred(int x) { long long re = 0,cnt = 0; for(int i = x;i;i -= i&-i) re += pred[i],cnt += cnt1[i]; return cnt * x - re; } inline long long GetSucc(int x) { long long re = 0,cnt = 0; for(int i = x;i < MAX;i += i&-i) re += succ[i],cnt += cnt2[i]; return re - cnt * x; }
POJ 1990 MooFest 树状数组
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。