Median of an Array(贪心策略,编程技巧)

文章目录

题目描述

给你一个由 n n n 个整数组成的数组 a a a 。

数组 q 1 , q 2 , ... , q k q_1,q_2,...,q_k q1,q2,...,qk 的中位数是 p ⌈ k 2 ⌉ p⌈\frac {k}{2}⌉ p⌈2k⌉ ,其中 p p p 是按非递减顺序排列的数组 q q q 。

例如,数组 [ 9 , 5 , 1 , 2 , 6 ] [9,5,1,2,6] [9,5,1,2,6] 的中位数是 5 5 5 ,在排序数组 [ 1 , 2 , 5 , 6 , 9 ] [1,2,5,6,9] [1,2,5,6,9] 中,索引 ⌈ 5 2 ⌉ = 3 ⌈\frac {5}{2}⌉=3 ⌈25⌉=3 处的数字是 5 5 5 ;数组 [ 9 , 2 , 8 , 3 ] [9,2,8,3] [9,2,8,3] 的中位数是 3 3 3 ,在排序数组 [ 2 , 3 , 8 , 9 ] [2,3,8,9] [2,3,8,9] 中,索引 ⌈ 4 2 ⌉ = 2 ⌈\frac {4}{2}⌉=2 ⌈24⌉=2 处的数字是 3 3 3 。

您可以选择一个整数 i ( 1 ≤ i ≤ n ) i( 1≤i≤n) i(1≤i≤n),并在一次操作中将 a i a_i ai 增加 1 1 1 。

你的任务是找出增加数组中位数所需的最少运算次数。

注意数组 a a a 不一定包含不同的数。

输入格式

第一行包含一个整数 n ( 1 ≤ n ≤ 1 0 5 ) n(1 \leq n \leq 10^5) n(1≤n≤105)- 数组 a a a 的长度。

第二行包含 n n n 个整数 a 1 , a 2 , . . . , a n ( 1 ≤ a i ≤ 1 0 9 ) a_1,a_2,...,a_n(1 \leq a_i \leq 10^9) a1,a2,...,an(1≤ai≤109) - 数组 a a a 。

输出格式

输出一个整数 - 增加数组中位数所需的最少操作数。

样例输入

cpp 复制代码
3
2 2 8

样例输出

cpp 复制代码
1

提交链接

https://hydro.ac/d/lp728/p/12

提示

样例解释 1 1 1:

对第一个数字进行一次运算,得到数组 [ 3 , 2 , 8 ] [3,2,8] [3,2,8],这个数组的中位数是 3 3 3,因为它是非递减排序数组 [ 2 , 3 , 8 ] [2,3,8] [2,3,8] 中索引 ⌈ 3 2 ⌉ = 2 ⌈\frac {3}{2}⌉=2 ⌈23⌉=2 处的数字。原数组 [ 2 , 2 , 8 ] [2,2,8] [2,2,8] 的中位数是 2 2 2,因为它是非递减排序数组 [ 2 , 2 , 8 ] [2,2,8] [2,2,8] 中索引 ⌈ 3 2 ⌉ = 2 ⌈\frac {3}{2}⌉=2 ⌈23⌉=2 处的数字。因此,只需一次操作,中位数就增加了。 ( 3 > 2 ) (3>2) (3>2)

解析

先对数组进行排序,找出数组中的中位数,即数字 a ⌈ n 2 ⌉ a_{⌈\frac {n}{2}⌉} a⌈2n⌉,让它等于 x x x 。为了使中位数增加,即至少变为 x + 1 x+1 x+1,数组中必须至少有 n − ⌈ n 2 ⌉ + 1 n-⌈\frac {n}{2}⌉+1 n−⌈2n⌉+1 个数字大于或等于 x + 1 x+1 x+1。统计 x x x 的个数即可。

count 函数:统计区间内某个数的个数。

参考代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
int t , n;
int main()
{
	int n;
	cin >> n;
	vector<int> v(n);   
	for(int i = 0; i < n; i++)
		cin >> v[i];
	sort(v.begin() , v.end());  //排序 
	int id = (n + 1) / 2 - 1;  //中位数的下标(下标从0输入) 
	int num = count(v.begin() + id , v.end() , v[id]);  //计算v[id]的个数 
	cout << num << endl;
	return 0;
}
相关推荐
落魄君子8 分钟前
SVM分类-支持向量机(Support Vector Machine)
神经网络·算法·支持向量机·分类
小王爱吃月亮糖14 分钟前
补充--C++的项目结构和管理
数据结构·c++·笔记·学习
上理考研周导师1 小时前
【单片机原理】第1章 微机基础知识,运算器,控制器,寄存器,微机工作过程,数制转换
算法
IT猿手2 小时前
基于PWLCM混沌映射的麋鹿群优化算法(Elk herd optimizer,EHO)的多无人机协同路径规划,MATLAB代码
算法·elk·机器学习·matlab·无人机·聚类·强化学习
m0_675988237 小时前
Leetcode2545:根据第 K 场考试的分数排序
python·算法·leetcode
破-风7 小时前
leetcode---mysql
算法·leetcode·职场和发展
Wils0nEdwards7 小时前
Leetcode 合并两个有序链表
算法·leetcode·链表
eternal__day9 小时前
数据结构十大排序之(冒泡,快排,并归)
java·数据结构·算法
姚先生979 小时前
LeetCode 35. 搜索插入位置 (C++实现)
c++·算法·leetcode
Theodore_102210 小时前
3 需求分析
java·开发语言·算法·java-ee·软件工程·需求分析·需求