牛客NC404 最接近的K个元素【中等 二分查找+双指针 Java/Go/PHP】

题目

题目链接:

https://www.nowcoder.com/practice/b4d7edc45759453e9bc8ab71f0888e0f

知识点

复制代码
	二分查找;找到第一个大于等于x的数的位置idx;
	然后从idx开始往两边扩展

Java代码

java 复制代码
import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param nums int整型ArrayList
     * @param k int整型
     * @param x int整型
     * @return int整型ArrayList
     */
    public ArrayList<Integer> closestElement (ArrayList<Integer> nums, int k,
            int x) {
        //二分+双指针
        int n = nums.size();
        int right = firstGt(nums, x);

        int left = right - 1;

        ArrayList<Integer> ans = new ArrayList<>();
        LinkedList<Integer> ll = new LinkedList<>();
        if (right == 0) {
            for (int i = 0; i < k ; i++) {
                ll.add(nums.get(i));
            }
        } else if (right == n - 1) {
            for (int i = n - 1 - k; i < n ; i++) {
                ll.add(nums.get(i));
            }
        } else {
            while (left >= 0 || right < n) {
                int diffleft = -1;
                int diffright = -1;

                if (left >= 0) {
                    diffleft = x - nums.get(left);
                }
                if (right < n) {
                    diffright = nums.get(right) - x;
                }

                if (diffleft != -1 && diffright != -1) {

                    if (diffleft <= diffright) {
                        ll.addFirst(nums.get(left--));
                    } else {
                        ll.addLast(nums.get(right++));
                    }

                } else if (diffleft != -1) {

                    ll.addFirst(nums.get(left--));
                } else if (diffright != -1) {

                    ll.addLast(nums.get(right++));
                }

                if (ll.size() == k)
                    break;
            }
        }

        return new ArrayList<>(ll);
    }

    //找到大于等于x的下标位置
    public int firstGt(ArrayList<Integer> nums, int x) {
        int n = nums.size();
        int left = 0;
        int right = n;
        while (left < right) {
            int m = left + (right - left) / 2;
            if (nums.get(m) > x) {
                right = m - 1;
            } else if (nums.get(m) < x) {
                left = m + 1;
            } else {
                return m;
            }
        }

        return left;
    }
}

Go代码

go 复制代码
package main

//import "fmt"

/**
 * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
 *
 *
 * @param nums int整型一维数组
 * @param k int整型
 * @param x int整型
 * @return int整型一维数组
 */
func closestElement(nums []int, k int, x int) []int {
	//二分+双指针
	n := len(nums)
	right := firstGt(nums, x)

	arrleft := []int{}
	arrright := []int{}
	ans := []int{}

	if right == 0 {
		for i := 0; i < k; i++ {
			ans = append(ans, nums[i])
		}
	} else if right == n-1 {
		for i := n - 1 - k; i < n; i++ {
			ans = append(ans, nums[i])
		}
	} else {
		left := right - 1
		cnt := 0
		for left >= 0 || right < n {
			diffleft := -1
			diffright := -1

			if left >= 0 {
				diffleft = x - nums[left]
			}
			if right < n {
				diffright = nums[right] - x
			}

			if diffleft != -1 && diffright != -1 {
				if diffleft <= diffright {
					arrleft = append(arrleft, nums[left])
					left--
				} else {
					arrright = append(arrright, nums[right])
					right++
				}
			} else if diffleft != -1 {
				arrleft = append(arrleft, nums[left])
				left--
			} else if diffright != -1 {
				arrright = append(arrright, nums[right])
				right++
			}

			cnt++
			if cnt == k {
				for i := len(arrleft) - 1; i >= 0; i-- {
					ans = append(ans, arrleft[i])
				}
				for i := 0; i < len(arrright); i++ {
					ans = append(ans, arrright[i])
				}

				break
			}
		}
	}
	return ans
}

//找到大等于x的位置
func firstGt(nums []int, x int) int {
	n := len(nums)
	left := 0
	right := n - 1

	for left < right {
		m := left + (right-left)/2

		if nums[m] > x {
			right = m - 1
		} else if nums[m] < x {
			left = m + 1
		} else {
			return m
		}
	}

	return left
}

PHP代码

php 复制代码
<?php


/**
 * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
 *
 * 
 * @param nums int整型一维数组 
 * @param k int整型 
 * @param x int整型 
 * @return int整型一维数组
 */
function closestElement( $nums ,  $k ,  $x )
{
      // 二分+双指针
    $n = count($nums);
    $right = firstGt($nums,$x);

    $ans = [];
    $arrleft=[];
    $arrright =[];

    if($right ==0 ){
        for($i=0;$i<$k;$i++){
            $ans[$i] = $nums[$i];
        }
    }else if($right ==$n-1){
        for($i=$n-1-$k;$i>=0;$i++){
            $ans[count($ans)] = $nums[$i];
        }
    }else {
        $left = $right-1;
        $cnt =0;
        while ($left>=0 || $right < $n){
            $diffleft=-1;
            $diffright =-1;

            if($left>=0) {
                $diffleft = $x-$nums[$left];
            }

            if($right<$n){
                $diffright = $nums[$right]-$x;
            }

            if($diffleft!=-1 && $diffright!=-1){
                if($diffleft<=$diffright){
                    $arrleft[count($arrleft)] = $nums[$left--];
                }else{
                    $arrright[count($arrright)] = $nums[$right++];
                }

            }else if($diffleft!=-1){
                $arrleft[count($arrleft)] = $nums[$left--];
            }else if($diffright!=-1){
                $arrright[count($arrright)] = $nums[$right++];
            }

            $cnt++;
            if($cnt==$k){
                for($i=count($arrleft)-1;$i>=0;$i--){
                    $ans[count($ans)] = $arrleft[$i];
                }
                for($i=0;$i<count($arrright);$i++){
                    $ans[count($ans)] = $arrright[$i];
                }

                break;
            }
        }
    }

    return $ans;
}


//找到大于等于x的位置
function firstGt($nums,$x){
    $n = count($nums);
    $left =0;
    $right = $n;
    while ($left<$right){
        $m = $left+(($right-$left)>> 1);
        if($nums[$m] > $x){
            $right = $m-1;
        }else if($nums[$m] < $x){
            $left = $m+1;
        }else{
            return $m;
        }
    }
    return $left;
}
相关推荐
IT从业者张某某3 分钟前
机器学习-04-分类算法-03KNN算法案例
算法·机器学习·分类
chen_song_6 分钟前
WebRTC的ICE之TURN协议的交互流程中继转发Relay媒体数据的turnserver的测试
算法·音视频·webrtc·交互·媒体
蒙奇D索大22 分钟前
【数据结构】图解图论:度、路径、连通性,五大概念一网打尽
数据结构·考研·算法·图论·改行学it
uhakadotcom26 分钟前
2025年春招:如何使用DeepSeek + 豆包优化简历,轻松敲开心仪公司的大门
算法·面试·github
小白狮ww30 分钟前
Retinex 算法 + MATLAB 软件,高效率完成图像去雾处理
开发语言·人工智能·算法·matlab·自然语言处理·图像识别·去雾处理
trust Tomorrow2 小时前
每日一题-力扣-2278. 字母在字符串中的百分比 0331
算法·leetcode
Lecea_L2 小时前
你能在K步内赚最多的钱吗?用Java解锁最大路径收益算法(含AI场景分析)
java·人工智能·算法
Tony882 小时前
热题100 - 394. 字符串解码
java·算法
Lecea_L2 小时前
🔍 找到数组里的“节奏感”:最长等差子序列
java·算法
是Dream呀2 小时前
ResNeXt: 通过聚合残差变换增强深度神经网络
人工智能·算法