树状数组 POJ1990 MOOFEST

来补一补数据结构的题，当时这道题完全不懂，连题意都理解了半天，今天又花了两个小时，终于搞懂啦！感觉不懂这道题的时候怎么看人家的博客都感觉难以理解，弄懂了之后，看到每篇博客都感觉明明白白~~~

题目链接

题目大意：

第一行给出牛的个数n,接下来有n行，分别代表着这n头牛的音量承受值和位置，这些牛直线排列，每两头牛之间都要对话，每一对牛耗费的音量都为：这一对牛中的最大音量承受值*这两头牛之间的距离，求所有对牛的音量总和。

鉴于上一篇博客颤抖的直线，这一次我选择徒手作图：

 

 

ACcode:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
 
typedef long long int64;
const int MAXN = 20010;
 
struct Node{
    int x;
    int v;
    bool operator<(const Node& s)const{
        return v < s.v; 
    }
};
Node node[MAXN];
int n;
int treeCount[MAXN];
int treeDis[MAXN];
 
int lowbit(int x){
    return x&(-x);
}
 
int64 getSum(int x , int *arr){
    int64 sum = 0;
    while(x){
         sum += arr[x];
         x -= lowbit(x);
    }
    return sum;
}
 
void add(int x , int val , int *arr){
    while(x < MAXN){
         arr[x] += val; 
         x += lowbit(x); 
    }
}
 
int64 solve(){
    int64 ans = 0;
    int64 totalDis = 0;
    memset(treeCount , 0 , sizeof(treeCount));
    memset(treeDis , 0 , sizeof(treeDis));
    sort(node , node+n);
 
    for(int i = 0 ; i < n ; i++){
        int64 count = getSum(node[i].x , treeCount); 
        int64 dis = getSum(node[i].x , treeDis);
        ans += node[i].v*(count*node[i].x-dis);
        ans += node[i].v*((totalDis-dis-(i-count)*node[i].x));
        totalDis += node[i].x;
        add(node[i].x , 1 , treeCount);
        add(node[i].x , node[i].x , treeDis);
    }
    return ans;
}
 
int main(){
    while(scanf("%d" , &n) != EOF){
         for(int i = 0 ; i < n ; i++) 
             scanf("%d%d" , &node[i].v , &node[i].x);
         printf("%lld\n" , solve());
    }
    return 0;
}