AcWing 1215. 小朋友排队 (树状数组)

2024-03-27 22:04

本文主要是介绍AcWing 1215. 小朋友排队 (树状数组),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

n n n 个小朋友站成一排。

现在要把他们按身高从低到高的顺序排列,但是每次只能交换位置相邻的两个小朋友。

每个小朋友都有一个不高兴的程度。

开始的时候,所有小朋友的不高兴程度都是 0 0 0

如果某个小朋友第一次被要求交换,则他的不高兴程度增加 1 1 1,如果第二次要求他交换,则他的不高兴程度增加 2 2 2(即不高兴程度为 3 3 3),依次类推。当要求某个小朋友第 k k k 次交换时,他的不高兴程度增加 k k k

请问,要让所有小朋友按从低到高排队,他们的不高兴程度之和最小是多少。

如果有两个小朋友身高一样,则他们谁站在谁前面是没有关系的。

输入格式
输入的第一行包含一个整数 n n n,表示小朋友的个数。

第二行包含 n n n 个整数 H 1 , H 2 , … , H n H_1,H_2,…,H_n H1,H2,,Hn,分别表示每个小朋友的身高。

输出格式
输出一行,包含一个整数,表示小朋友的不高兴程度和的最小值。

数据范围
1 ≤ n ≤ 100000 , 1≤n≤100000, 1n100000,
0 ≤ H i ≤ 1000000 0≤H_i≤1000000 0Hi1000000
输入样例:

3
3 2 1

输出样例:

9

样例解释
首先交换身高为 3 3 3 2 2 2 的小朋友,再交换身高为 3 3 3 1 1 1 的小朋友,再交换身高为 2 2 2 1 1 1 的小朋友,每个小朋友的不高兴程度都是 3 3 3,总和为 9 9 9


首先明确只能相邻的人之间换,而且每次换的时候,比如 1 1 1 2 2 2 换,那么他们两个人的不高兴都会加上他们各自交换过的次数。

对于样例我们只看中间的 2,2 的左边有一个 3,这个 3 应该在 2 的右边,2 的右边有一个 1,这个 1 应该在 2 的左边,不管怎么换 2 都会被换两次,因为如果 2 先和 3 换,3 和 1 换,1 和 2 换, 2 换了两次,如果 2 先和 1 换, 3 和 1 换,3 和 2 换,2 还是换了两次。

对于 1 也是,因为左边有两个比他大的,所以 1 要换两次。

对于任何一个数,如果左边有比他大的,就一定要跨过他去,如果右边有比他小的,也一定要跨过他去,这么下来就交换了左边比他大的个数加右边比他小的个数的次数,这样得来的就已经是最小的交换次数了,因为不可能比这个小的情况,如果有就不满足升序了。

那么我们就需要实现求出来某个数左边比他大的数的个数和右边比他小的数的个数。可以有很多方法,这里使用树状数组实现。

这里树状数组代表的是某个数有没有,如果有就是 1,没有就是 0 。
之后如果要求a[i]的左边比它大的数,只要求从左边扫过去,然后求出最大高度的前缀和减去a[i]的前缀和
求右边的比当前数小的数则是从右边扫,其他的同理。

此种做法是树状数组的一种管用用法,可以参考树状数组的笔记

代码:

#include<iostream>
#include<cstring>
using namespace std;
const int N = 1e5 + 10;
const int M = 1e6 + 10;int tr[M];
int a[N];
int n;
int times[N];	//存每个数的交换个数int lowbit(int x) { return x & -x; }
void add(int x, int c) {for (int i = x; i <= M; i+=lowbit(i))tr[i] += c;
}
int sum(int x) {int res = 0;for (int i = x; i; i -= lowbit(i))res += tr[i];return res;
}int main() {cin >> n;for (int i = 1; i <= n; i++) {cin >> a[i];a[i] ++;}for (int i = 1; i <= n; i++) {times[i] = sum(M) - sum(a[i]);add(a[i], 1);}memset(tr, 0, sizeof tr);for (int i = n; i >= 1; i--) {times[i] += sum(a[i] - 1);add(a[i], 1);}long long res = 0;for (int i = 1; i <= n; i++) {res += (long long)(1 + times[i]) * times[i] / 2;	//注意根据题目条件,答案要求等差数列的和}cout << res;return 0;
}

这篇关于AcWing 1215. 小朋友排队 (树状数组)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/853519

相关文章

C++原地删除有序数组重复项的N种方法

《C++原地删除有序数组重复项的N种方法》给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度,不要使用额外的数组空间,你必须在原地修改输入数组并在使用O(... 目录一、问题二、问题分析三、算法实现四、问题变体:最多保留两次五、分析和代码实现5.1、问题分析5.

Java中数组转换为列表的两种实现方式(超简单)

《Java中数组转换为列表的两种实现方式(超简单)》本文介绍了在Java中将数组转换为列表的两种常见方法使用Arrays.asList和Java8的StreamAPI,Arrays.asList方法简... 目录1. 使用Java Collections框架(Arrays.asList)1.1 示例代码1.

C++一个数组赋值给另一个数组方式

《C++一个数组赋值给另一个数组方式》文章介绍了三种在C++中将一个数组赋值给另一个数组的方法:使用循环逐个元素赋值、使用标准库函数std::copy或std::memcpy以及使用标准库容器,每种方... 目录C++一个数组赋值给另一个数组循环遍历赋值使用标准库中的函数 std::copy 或 std::

C++初始化数组的几种常见方法(简单易懂)

《C++初始化数组的几种常见方法(简单易懂)》本文介绍了C++中数组的初始化方法,包括一维数组和二维数组的初始化,以及用new动态初始化数组,在C++11及以上版本中,还提供了使用std::array... 目录1、初始化一维数组1.1、使用列表初始化(推荐方式)1.2、初始化部分列表1.3、使用std::

C++ Primer 多维数组的使用

《C++Primer多维数组的使用》本文主要介绍了多维数组在C++语言中的定义、初始化、下标引用以及使用范围for语句处理多维数组的方法,具有一定的参考价值,感兴趣的可以了解一下... 目录多维数组多维数组的初始化多维数组的下标引用使用范围for语句处理多维数组指针和多维数组多维数组严格来说,C++语言没

Java 字符数组转字符串的常用方法

《Java字符数组转字符串的常用方法》文章总结了在Java中将字符数组转换为字符串的几种常用方法,包括使用String构造函数、String.valueOf()方法、StringBuilder以及A... 目录1. 使用String构造函数1.1 基本转换方法1.2 注意事项2. 使用String.valu

JAVA中整型数组、字符串数组、整型数和字符串 的创建与转换的方法

《JAVA中整型数组、字符串数组、整型数和字符串的创建与转换的方法》本文介绍了Java中字符串、字符数组和整型数组的创建方法,以及它们之间的转换方法,还详细讲解了字符串中的一些常用方法,如index... 目录一、字符串、字符数组和整型数组的创建1、字符串的创建方法1.1 通过引用字符数组来创建字符串1.2

vue如何监听对象或者数组某个属性的变化详解

《vue如何监听对象或者数组某个属性的变化详解》这篇文章主要给大家介绍了关于vue如何监听对象或者数组某个属性的变化,在Vue.js中可以通过watch监听属性变化并动态修改其他属性的值,watch通... 目录前言用watch监听深度监听使用计算属性watch和计算属性的区别在vue 3中使用watchE

hdu2241(二分+合并数组)

题意:判断是否存在a+b+c = x,a,b,c分别属于集合A,B,C 如果用暴力会超时,所以这里用到了数组合并,将b,c数组合并成d,d数组存的是b,c数组元素的和,然后对d数组进行二分就可以了 代码如下(附注释): #include<iostream>#include<algorithm>#include<cstring>#include<stack>#include<que

hdu 1166 敌兵布阵(树状数组 or 线段树)

题意是求一个线段的和,在线段上可以进行加减的修改。 树状数组的模板题。 代码: #include <stdio.h>#include <string.h>const int maxn = 50000 + 1;int c[maxn];int n;int lowbit(int x){return x & -x;}void add(int x, int num){while