题目背景
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;
}