和1009一样的,直接套我1009的模板就行,嘻嘻
#include <iostream>
using namespace std;
typedef long long ll;
// ================================= Segment Tree =================================
const int maxn = 1e6 + 5;
struct Node{
int l,r,f,val; // val->区间和 f->懒惰标记,区间加几f就等于几 l,r->对应的区间为[l,r]
}sgt[maxn * 4];
inline int ls(int k){return k << 1;} // 左孩子
inline int rs(int k){return k << 1 | 1;} // 右孩子
inline void pushup(int k){sgt[k].val = (sgt[ls(k)].val + sgt[rs(k)].val);} // 该节点的val = 左孩子的val + 右孩子的val
inline void pushdown(int k){
sgt[ls(k)].f += sgt[k].f;
sgt[rs(k)].f += sgt[k].f;
sgt[ls(k)].val += (sgt[ls(k)].r - sgt[ls(k)].l + 1) * sgt[k].f; // 整个区间的和要+=区间元素的个数*f
sgt[rs(k)].val += (sgt[rs(k)].r - sgt[rs(k)].l + 1) * sgt[k].f;
sgt[k].f = 0;
}
void build(int l,int r,int k = 1){ // 原数组中[l,r]以sgt[k]为根build线段树
sgt[k].l = l,sgt[k].r = r;
if(l == r){
sgt[k].val = 0;
return;
}
int m = (l + r) >> 1;
build(l,m,ls(k)); // 递归建树
build(m + 1,r,rs(k));
pushup(k);
}
void modify(int x,int y,int z,int k = 1){ // 令[x,y]区间内数值全部加z
int l = sgt[k].l,r = sgt[k].r;
if(x <= l && y >= r){
sgt[k].f += z; // 懒惰标记,表示sgt[k]的子树内的元素全部加z
sgt[k].val += (r - l + 1) * z; // 更新sgt[k]的部分和
return;
}
if(sgt[k].f) pushdown(k); // 若这个区间不完全包括要改的区间,先pushdown
int m = (l + r) >> 1;
if(x <= m) modify(x,y,z,ls(k));
if(y > m) modify(x,y,z,rs(k));
pushup(k);
}
int query(int x,int y,int k = 1){ // 查询[x,y]区间内数值之和
int l = sgt[k].l,r = sgt[k].r;
if(x <= l && y >= r) return sgt[k].val;
if(sgt[k].f) pushdown(k);
int sum = 0,m = (l + r) >> 1;
if(x <= m) sum += query(x,y,ls(k));
if(y > m) sum += query(x,y,rs(k));
return sum;
}
// ================================= Segment Tree =================================
int smaller[maxn]; // 这个数前面比它大的数的个数 & 这个数后面比它小的数的个数
int a[maxn]; // 原始数组
int seq[maxn]; // 离散化后数组
// ================================== Discretize ==================================
struct N{
int val,order;
}temp[maxn];
#include <algorithm>
bool cmp(N a,N b){
return a.val < b.val;
}
void discretize(int n){ // 离散化
for(int i = 1;i <= n;i++){
temp[i].val = a[i];
temp[i].order = i;
}
sort(temp + 1,temp + n + 1,cmp);
for(int i = 1;i <= n;i++){
seq[temp[i].order] = i;
}
}
// ================================== Discretize ==================================
// 逆序对数
void inverseNum(int n){
build(1,n);
for(int i = n;i >= 1;i--){
modify(seq[i],seq[i],1); // [seq[i],seq[i]]区间加一
smaller[i] = query(1,seq[i] - 1); // 区间内数值之和
}
}
int main(){
int n;
scanf("%d",&n);
for(int i = 1;i <= n;i++){
scanf("%d",&a[i]);
}
discretize(n);
inverseNum(n);
for(int i = 1;i <= n;i++){
printf("%d",smaller[i]);
if(i != n) printf(" ");
}
system("pause");
return 0;
}