IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 数据结构与算法 -> LeetCode658. 找到K个最近的元素 -> 正文阅读

[数据结构与算法]LeetCode658. 找到K个最近的元素


思路

减法思路吧
题目如果是找一个最近的数,那找到X在哪就好了。
但是X不一定在数组内,所以要对二分法改一点。

一般的二分法,就是返回找到后的索引,找不到就返回-1。
所以这里可以变成:找到了就返回索引,找不到就返回左右指针(就是x左边的数和x右边的数了)。

到这里,就找到X(或者X的邻居)了。
接下来用双指针扫描就好了:

1.两个指针L, R指向X的位置或者它的邻居
2.按照题目要求往左或者往右移动指针
3.结束条件就是左右指针差为K

总的来说,思路如上,不过玛德好多坑啊,可能是太菜了,好多边界条件,可能会有简化版本,欢迎讨论。


代码

public class FindKClosestElements {
	
	public int[] binarySearch(int[] arr, int l, int r, int x) {
		if(l==r) {
			return new int[] {l};
		}
				
		int m=l+(r-l)/2;
		
		if(x>arr[m]) {
			l=m+1;
			if(l>arr.length-1) {
				return new int[] {arr.length-1};
			}
			if(arr[l]>x) {
				return new int[] {l-1,l};
			}
		}else if(x<arr[m]) {
			r=m-1;
			if(r<0) {
				return new int[] {0};
			}
			if(arr[r]<x) {
				return new int[] {r,r+1};
			}
		}else {
			return new int[] {m};
		}
		int[] res=binarySearch(arr,l,r,x);
		return res;
	}
	
	 public List<Integer> findClosestElements(int[] arr, int k, int x) {
	        int[] nearest=binarySearch(arr, 0, arr.length-1, x);
	        int l=nearest[0];
	        int r=nearest[0];

	        if(nearest.length>=2) {
	        	if(arr[nearest[1]]-x<x-arr[nearest[0]]) {
	        		l=nearest[1];
	        		r=nearest[1];
	        	}
	        }
	        while(r-l<k-1&&!(l==0&&r==arr.length-1)) {
	        	if(l==0) {
	        		r++;
	        		continue;
	        	}
	        	if(r==arr.length-1) {
	        		l--;
	        		continue;
	        	}
	        	if(arr[r+1]-x<x-arr[l-1]) {
	        		r++;
	        		continue;
	        	}else {
	        		l--;
	        	}
	        }
	        
	        List<Integer> res=new ArrayList<Integer>();
	        for(int i=l;i<=r;i++) {
	        	res.add(arr[i]);
	        }
	        return res;
	        
	    }
	 
	 public static void main(String[] args) {
		int[] arr=new int[] {1,1,1,10,10,10};
		System.out.println(new FindKClosestElements().findClosestElements(arr, 1, 9));
	}
}
  数据结构与算法 最新文章
【力扣106】 从中序与后续遍历序列构造二叉
leetcode 322 零钱兑换
哈希的应用:海量数据处理
动态规划|最短Hamilton路径
华为机试_HJ41 称砝码【中等】【menset】【
【C与数据结构】——寒假提高每日练习Day1
基础算法——堆排序
2023王道数据结构线性表--单链表课后习题部
LeetCode 之 反转链表的一部分
【题解】lintcode必刷50题<有效的括号序列
上一篇文章      下一篇文章      查看所有文章
加:2021-07-05 20:27:23  更:2021-07-05 20:27:57 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年4日历 -2024/4/26 23:48:01-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码