每日算法:洛谷U535992 J-C 小梦的宝石收集(双指针、二分)

题目描述

小梦有 n 颗能量宝石,其中第 i 颗的能量为 ai​,但这些能量宝石十分不稳定,随时有可能发生崩坏,导致他们全部消失!

小梦想要留住宝石们,不希望他们发生崩坏,同时他发现:如果这些宝石的能量的极差越大,则这些宝石们越不稳定,因此他希望最小化这些宝石的能量的极差(最大值减去最小值的差值)。

小梦有一招点石成金的技能,这个技能可以以如下方式改变一些宝石的能量:

∙ 要么小梦选择一块能量最小的宝石,将他变成一块当前能量最大的宝石。

∙ 要么小梦选择一块能量最大的宝石,将他变成一块当前能量最小的宝石。

形式化的即:

∙ 选择 i (1≤i≤n,ai​=min(a1​,a2​,...,an​)),执行:ai​:=max(a1​,a2​,...,an​)。

∙ 或选择 i (1≤i≤n,ai​=max(a1​,a2​,...,an​)),执行:ai​:=min(a1​,a2​,...,an​)。

小梦至多可以使用 k 次上述的 "点石成金" 技能,他想知道这些宝石的能量极差最小可以变为多少,请你帮他算一算吧。

输入格式

本题有多组测试数据。

输入的第一行包含一个正整数 T,表示数据组数。

接下来包含 T 组数据,每组数据的格式如下:

第一行两个正整数 n,k,分别表示小梦拥有的宝石数量 n ,和他最多可以释放 "点石成金" 技能的次数 k。

第二行 n 个正整数 ai​,表示每块宝石的能量。

输出格式

对于每组测试数据:

在单独的一行输出一个整数,表示宝石能量的极差最小值。

输入输出样例

输入 #1

复制代码
2
8 3
1 2 3 4 5 6 7 8
4 3
100 1 100 2

输出 #1

复制代码
4
0

说明/提示

【样例 1 解释】

使用 3 次操作一,选择 min 变为 max,操作完后数组变成:{8,8,8,4,5,6,7,8}。

此时数组的极差为 4 最小,可以证明不存在比 4 更优的答案。

【数据范围】

令 N 表示 T 组数据中 n 的总和。

对于 30% 的数据有:T=1,1≤k≤N≤20。

对于 60% 的数据有:1≤T≤10,1≤k≤N≤2000。

对于所有的测试数据有: 1≤T≤1000,1≤N≤5×105,0≤k≤n,0≤ai​≤109。

思路:

我们很容易想到,先将数组进行排序,然后分别用两个指针l,和r表示最左侧的最大值和最右侧的最小值。

每次只要有将最小值变成最大值或者将最大值变成最小值的情况出现,那么我们就需要移动左或右指针到次小或次大值的位置上。

这样,我们假设当l指针在i位置,r指针在j位置时,找到答案,则操作次数为:(i-1)【操作左边要用的次数】+(n-j)【操作右边要用的次数】+min(i-1,n-j)【将左边的值变成最大值或将右边的值变为最小值后,额外再需要移动的次数 例:如果左边是1,右边是8,修改左边后,最后面的两个数应该是8,8因此如果此时要再操作右边,应该额外加上左边的操作次数,同理另一种情况也是如此,而这个值我们希望它尽可能小,这样可操作的次数就会变多】

而由题目规定,我们可知:(i-1)+(n-j)+min(i-1,n-j) <= k

因此当符合这一规定时,我们 则用ans记录它的差值,取最小值。

于是我们可以很轻松的想到双重循环遍历枚举,但是双重循环肯定过不了,然后我们发现:该数组中的数是单调递增的,且满足二分性,因此我们可以将第二重循环优化为二分,这样的时间复杂度是nlogn。

代码:

cpp 复制代码
#include<bits/stdc++.h>

using namespace std;
int t,n,k;
const int N = 5e5+10;
int q[N];

int main(){
	cin>>t;
	while(t--){
		cin>>n>>k;
		for(int i =1;i<=n;i++){
			cin>>q[i];
		}
		sort(q+1,q+n+1);//先从小到大排序 
		int ans = INT_MAX;//初始化为极大值 
	for(int i = 1;i<=n;i++){
		if(i - 1 > k) break;
		int l = i,r = n;//在从i到n的区间里找到右边界 
		if(i - 1>k) break;//操作次数超过k了,直接跳出 
		while(l <= r){
		int mid = (l+r) / 2;
		if((i-1) + (n - mid) + min(i-1,n-mid) <= k) r = mid-1;
		else l = mid+1;
		}
		ans = min(ans,q[l]-q[i]);//每次更新取最小值 
	}
	cout<<ans<<endl;
	}
	
return 0;
}
相关推荐
倔强的石头_几秒前
【数据结构与算法】深入理解 单链表
后端·算法
未来之窗软件服务25 分钟前
搭建 Select 三级联动架构-东方仙盟插件开发 JavaScript ——仙盟创梦IDE
开发语言·javascript·ide·仙盟创梦ide·东方仙盟皮肤·东方仙盟·东方仙盟插件
@曲终33 分钟前
C++:栈帧、命名空间、引用
java·开发语言·c++·经验分享·笔记
蓝婷儿34 分钟前
6个月Python学习计划 Day 8 - Python 函数基础
开发语言·python·学习
闫广庆40 分钟前
大模型与AI智能体方向面试经验分享
算法·架构
mit6.8241 小时前
[Rust_1] 环境配置 | vs golang | 程序运行 | 包管理
开发语言·学习·rust
青出于兰1 小时前
C语言| 函数参数传递指针
c语言·开发语言
黄雪超1 小时前
JVM——回顾:JVM的起源、特性与系统构成
java·开发语言·jvm
敲代码的瓦龙1 小时前
C++?多态!!!
c语言·开发语言·c++·windows·后端
我不是程序猿儿1 小时前
【C++】C++面向对象设计的核心思想之一: 接口抽象、解耦和可扩展性
java·开发语言·c++