[树状数组&离散化] CF 1311 F

题目

思路

数据比较大，先离散化。
可以先把x sort一遍通过v找出答案
如果x由小到大，前面v比当前V小d(i,j)=xi-xj;否则为0
i时前面所有的d为 比现在小的点数目*a[i].x-比现在小的点的距离和

代码

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
#define int long long
using namespace std;
const int INF = 0x3f3f3f3f;
const int MAXN=500010;
int b[500010];
int m,n;
struct node{
	int x,v;
	bool operator < (const node &b) const{
		return x<b.x; 
	}
}a[MAXN];
int lowbit(int x){
	return (-x)&x;
}
int bit[MAXN][3];
void edit(int pos,int val){
	for(int i=pos;i<=n;i+=lowbit(i)){
		bit[i][0]++;
		bit[i][1]+=val;
	}
}
int query(int pos,int k){
	int res=0;
	for(int i=pos;i>=1;i-=lowbit(i)){
		res+=bit[i][k];
	}
	return res;
}
signed main(){
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	//int n;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i].x;
	for(int i=1;i<=n;i++){
		cin>>a[i].v;
		b[i]=a[i].v;
	}
	sort(a+1,a+1+n);
	sort(b+1,b+1+n);
	m=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;i++) a[i].v=lower_bound(b+1,b+1+m,a[i].v)-b;
	int ans=0;
	for(int i=1;i<=n;i++){
		int num=query(a[i].v,0);
		int sum=query(a[i].v,1);
		ans+=num*a[i].x-sum;
		edit(a[i].v,a[i].x);
	}
	cout<<ans<<endl;
    return 0;
}