洛谷 P1428 小鱼比可爱 树状数组解法

题目描述

人比人，气死人；鱼比鱼，难死鱼。小鱼最近参加了一个“比可爱”比赛，比的是每只鱼的可爱程度。参赛的鱼被从左到右排成一排，头都朝向左边，然后每只鱼会得到一个整数数值，表示这只鱼的可爱程度，很显然整数越大，表示这只鱼越可爱，而且任意两只鱼的可爱程度可能一样。由于所有的鱼头都朝向左边，所以每只鱼只能看见在它左边的鱼的可爱程度，它们心里都在计算，在自己的眼力范围内有多少只鱼不如自己可爱呢。请你帮这些可爱但是鱼脑不够用的小鱼们计算一下。

输入格式

第一行输入一个整数 nn，表示鱼的数目。

第二行内输入 nn 个整数，用空格间隔，依次表示从左到右每只小鱼的可爱程度。

输出格式

行内输出 nn 个整数，用空格间隔，依次表示每只小鱼眼中有多少只鱼不如自己可爱。

输入输出样例

输入 #1复制

6
4 3 0 5 1 2

输出 #1复制

0 0 0 3 1 2

说明/提示

对于 100\%100% 的数据，n\leq 100n≤100。

 

第一次做树状数组的题目，记录一下模板，主要是对于原始数据需要进行离散化处理，方法是先按数字大小排序，然后把数字的值改为0~n或者1~n+1，再按下标排序，就可以用树状数组解决了

#include <iostream>
#include <algorithm>
using namespace std;
#define lowbit(x) ((x)&(-(x)))
int n;
int c[1000];
struct sa {
	int num;
	int idx;
}a[100];
bool cmp(const sa & a, const sa & b) {
	return a.num <= b.num;
}
bool cmp1(const sa & a, const sa & b) {
	return a.idx < b.idx;
}
void update(int x, const int & val) {
	while (x <= n) {
		c[x] += val;
		x += lowbit(x);
	}
}
int getSum(int x) {
	int ans = 0;
	while (x > 0) {
		ans += c[x];
		x -= lowbit(x);
	}
	return ans;
}
int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin >> n;
	for (int i = 0; i < n; ++i) {
		cin >> a[i].num;
		a[i].idx = i;
	}
	sort(a, a + n, cmp);
	for (int i = 0; i < n; ++i) {
		a[i].num = i;
	}
	sort(a, a + n, cmp1);
	for (int i = 0; i < n; ++i) {
		cout << getSum(a[i].num + 1) << ' ';
		update(a[i].num + 1, 1);
	}
	return 0;
}