华为OD-D卷小明找位置

小白踩坑2024-08-10 18:38

题目：小朋友出操，按学号从小到大排成一列；小明来迟了，请你给小明出个主意，让他尽快找到他应该排的位置。

算法复杂度要求不高于nLog(n)；学号为整数类型，队列规模<=10000；

输入描述:

1、第一行：输入已排成队列的小朋友的学号（正整数），以","隔开；

例如：93 95 97 100 102 123 155

2、第二行：小明学号，如110；

输出描述:

输出一个数字，代表队列位置（从1开始）。

例如：6

题目解析:

本题本质上还是对二分查找，只不过已知要找的数不在队列当中而已：

需要注意：队列位置（从1开始）

java 复制代码

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
//        int[] nums = new int[]{93, 95, 97, 100, 102, 123, 155};
//        int target = 110;
        // 处理数据
        Scanner scanner = new Scanner(System.in);
        String string = scanner.nextLine();
        String[] numsStr = string.split(",");
        int[] nums = new int[numsStr.length];
        for (int i = 0; i < nums.length; i++) {
            nums[i] = Integer.parseInt(numsStr[i]);
        }
        int target = scanner.nextInt();

        // 二分查找
        int left = 0;
        int right = nums.length - 1;
        int mid;
        // 注意等号，避免漏掉查找的数
        while (left <= right) {
            // 代替 (right+left) / 2 ,可以避免 right+left 超过 int 范围，一般情况下不用考虑
            mid = left + (right - left) / 2;
            if (target == nums[mid]) {
                System.out.println(mid + 1);
                return;
            } else if (target < nums[mid]) {
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        // 需要注意队列位置从1开始
        System.out.println(left + 1);
    }
}