POJ1990 MooFest，树状数组

最新推荐文章于 2022-02-23 15:02:25 发布

neofung

最新推荐文章于 2022-02-23 15:02:25 发布

阅读量1.2k

点赞数

分类专栏： POJ 树状数组文章标签： struct email c

本文链接：https://blog.csdn.net/neofung/article/details/7723256

版权

POJ 同时被 2 个专栏收录

105 篇文章 0 订阅

订阅专栏

树状数组

1 篇文章 0 订阅

订阅专栏

树状数组经典题，翻了一下别的解题报告，都是大同小异解释的不清楚，详情可以看我代码中的注释

/*******************************************************************************
 # Author : Neo Fung
 # Email : neosfung@gmail.com
 # Last modified: 2012-07-06 19:35
 # Filename: POJ1990 MooFest.cpp
 # Description : 
 ******************************************************************************/
#ifdef _MSC_VER
#define DEBUG
#define _CRT_SECURE_NO_DEPRECATE
#endif

#include <fstream>
#include <stdio.h>
#include <iostream>
#include <string.h>
#include <string>
#include <limits.h>
#include <algorithm>
#include <math.h>
#include <numeric>
#include <functional>
#include <ctype.h>
#define lowbit(x) (x&(-x))
using namespace std;

const int kMAX=20010;

// 树状数组中，count_sum统计的是对于第i头牛的坐标x，在它左边的牛的数目；
// pos_sum统计的是对于第i头牛的坐标x，在它左边的牛的坐标的和
int count_sum[kMAX],pos_sum[kMAX];

__int64 getsum(int x,const int *c)
{
  __int64 sum=0;
  for(int i=x;i>0;i-=lowbit(i))
      sum+=c[i];
  return sum;
}

void updata(int x,const int var,int *c)
{
  for(int i=x;i<kMAX;i+=lowbit(i))
      c[i]+=var;
}

struct NODE{
  int pos,var;
  inline bool operator<(const NODE &tmp) const
  {
    return var<tmp.var;
  }
}node[kMAX];

int main(void)
{
#ifdef DEBUG  
  freopen("../stdin.txt","r",stdin);
  freopen("../stdout.txt","w",stdout); 
#endif  

  int n;

  while(~scanf("%d",&n) && n)
  {
    for(int i=0;i<n;++i)
      scanf("%d%d",&node[i].var,&node[i].pos);

    sort(node,node+n);  //先对var从小到大排序，这样使得每次插入的牛都是所有牛中声音最大的
		memset(count_sum,0,sizeof(count_sum));
		memset(pos_sum,0,sizeof(pos_sum));

		__int64 ans=0;
		__int64 total=node[0].pos;
		updata(node[0].pos,1,count_sum);
		updata(node[0].pos,node[0].pos,pos_sum);

		for(int i=1;i<n;++i)
		{
			__int64 left_count=getsum(node[i].pos,count_sum);//在i号牛左边的牛的数目
			__int64 left_sum = getsum(node[i].pos,pos_sum);//在i号牛左边的所有牛的坐标和
			__int64 left_total = node[i].pos * left_count - left_sum;//在i号牛左边的所有牛离i号牛的距离和
			__int64 right_total = total - left_sum - (i-left_count)*node[i].pos;//在i号牛右边的所有牛离i号牛的距离和
			ans+=(left_total+right_total)*node[i].var;

			updata(node[i].pos,1,count_sum);
			updata(node[i].pos,node[i].pos,pos_sum);
			total+=node[i].pos;
		}

		printf("%I64d\n",ans);
  }

  return 0;
}