P2345 奶牛集会（树状数组）

题目背景

MooFest, 2004 Open

题目描述

约翰的N 头奶牛每年都会参加“哞哞大会”。哞哞大会是奶牛界的盛事。集会上的活动很

多，比如堆干草，跨栅栏，摸牛仔的屁股等等。它们参加活动时会聚在一起，第i 头奶牛的坐标为Xi，没有两头奶牛的坐标是相同的。奶牛们的叫声很大，第i 头和第j 头奶牛交流，会发出max{Vi; Vj}×|Xi − Xj | 的音量，其中Vi 和Vj 分别是第i 头和第j 头奶牛的听力。假设每对奶牛之间同时都在说话，请计算所有奶牛产生的音量之和是多少。

输入输出格式

输入格式：

• 第一行：单个整数N，1 ≤ N ≤ 20000

• 第二行到第N + 1 行：第i + 1 行有两个整数Vi 和Xi，1 ≤ Vi ≤ 20000; 1 ≤ Xi ≤ 20000

输出格式：

• 单个整数：表示所有奶牛产生的音量之和

模拟：

#include<iostream>
#include<cstdio>
#include<algorithm>
#define maxn 100001
using namespace std;
int n,m;
long long ans;
struct node{
    int v,x;
}e[100001];
int main() {
    scanf("%d",&n);
    for(int i=1;i<=n;++i) scanf("%d%d",&e[i].v,&e[i].x);
    for(int i=1;i<=n;++i) 
    for(int j=i+1;j<=n;++j)         //n方枚举累加。
    ans+=max(e[i].v,e[j].v)*abs(e[i].x-e[j].x);
    cout<<ans<<'\n';
    return 0;
}

思路：两个树状数组，cnt数组存他们坐标前有几头牛，sum数组存他们坐标前有牛的总坐标数。

#include<bits/stdc++.h>
#define INF 0x3fffffff
#define ll long long
#define mem(ar,num) memset(ar,num,sizeof(ar))
#define me(ar) memset(ar,0,sizeof(ar))
#define lowbit(x) ((x)&(-x))
#define IOS ios::sync_with_stdio(false)
#define DEBUG cout<<endl<<"DEBUG"<<endl;
using namespace std;
int n;
struct H{
    ll x,v;
}st[20010];
ll cnt[20009],sum[20009];//i-lowbit(i) ~ i范围内数的个数和这些数的和。cnt和sum是以他们坐标为标准
ll ans,num,s,maxn,tot,zh;
bool cmp(H a,H b){return a.v<b.v;}
void add(int x){
    for(int i=x;i<=maxn;i+=lowbit(i))
        cnt[i]+=1,sum[i]+=x;
}
void get(int x){//得到
    num=0,s=0;
    for(int i=x;i>=1;i-=lowbit(i))
        num+=cnt[i],s+=sum[i];//num是小于此声音牛的坐标号有几只牛,s是他们x的数目
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%lld%lld",&st[i].v,&st[i].x),maxn=max(maxn,st[i].x);
    sort(st+1,st+n+1,cmp);
    for(int i=1;i<=n;i++){
        zh+=st[i].x;//此声音之前所有牛位置和
        add(st[i].x);
        get(st[i].x);
        ans+=(num*st[i].x-s)*st[i].v;
        num=i-num;
        s=zh-s;
        ans+=(s-num*st[i].x)*st[i].v;
    }
    printf("%lld",ans);
    return 0;
}