原题链接:蓝桥杯历届试题-小朋友排队
解题思路:
这道题最终要求解的其实就是给定整数串中的逆序对的个数,但此题直接暴力是会超时的,所以利用树状数组进行求解,思路如下:首先利用输入的整数串构建一个树状数组,然后树状数组的getSum()可以求解从1开始的一些项的和,如果按顺序输入的话,就可以求解当前输入项之前比它矮的小朋友个数,而且也知道已经输入了多少个小朋友的身高,那么很自然可以求出大于等于当前身高的小朋友个数,但是还存在身高相等的情况,需要减去等于当前身高的个数,也就是getSum(i+1)-getSum(i),存入一个数组,然后,再从尾开始遍历,也可以用同样的方法,求取身高小于当前身高的小朋友个数,然后与之前的大于的对应相加,即可求得对于每一个身高,其逆序对的个数,最后将其全部加起来即是最终结果。
注意事项:
1. 要注意爆int,对于不高兴度,当交换次数很大的时候是会爆int的。
参考代码:
#include <iostream>
#include <cstring>
#define maxn 100010
#define MAXH 1000010
using namespace std;
long long total[maxn];
int c1[MAXH], c2[MAXH];
int num[maxn];
int cou[MAXH];
int n;
void init() {
total[0] = 0;
for (int i = 1; i <= maxn; i++) {
total[i] = total[i - 1] + i;
}
}
int lowbit(int x) {
return x & (-x);
}
void update(int pos, int val, int* c) {
while (pos <= MAXH) {
c[pos] += val;
pos += lowbit(pos);
}
}
int getSum(int pos, int* c) {
int ans = 0;
while (pos > 0) {
ans += c[pos];
pos -= lowbit(pos);
}
return ans;
}
int main() {
init();
while (cin >> n) {
memset(c1, 0, sizeof(c1));
memset(c2, 0, sizeof(c2));
memset(cou, 0, sizeof(cou));
for (int i = 1; i <= n; i++) {
cin >> num[i];
update(num[i] + 1, 1, c1);
cou[i] = i - getSum(num[i], c1);
cou[i] -= (getSum(num[i] + 1, c1) - getSum(num[i], c1));
}
for (int i = n; i > 0; i--) {
update(num[i] + 1, 1, c2);
cou[i] += getSum(num[i], c2);
}
long long ans = 0;
for (int i = 1; i <= n; i++) {
ans += total[cou[i]];
}
cout << ans << endl;
}
return 0;
}0.0分
1 人评分
C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:
一点编程也不会写的:零基础C语言学练课程
解决困扰你多年的C语言疑难杂症特性的C语言进阶课程
从零到写出一个爬虫的Python编程课程
只会语法写不出代码?手把手带你写100个编程真题的编程百练课程
信息学奥赛或C++选手的 必学C++课程
蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程
手把手讲解近五年真题的蓝桥杯辅导课程
发表评论 取消回复