万物的算法日记|算法学习 [java 二分法 acwing789.数的范围]

请添加图片描述

算法语言:java
题目来源:acwing.789

题目描述

给定一个按照升序排列的长度为 的整数数组,以及 q 个查询。

对于每个查询,返回一个元素 kk 的起始位置和终止位置(位置从 0 开始计数)。

如果数组中不存在该元素,则返回 -1 -1。

输入格式

第一行包含整数 n 和 q,表示数组长度和询问个数。

第二行包含 n 个整数(均在 1∼100001∼10000 范围内),表示完整数组。

接下来 q 行,每行包含一个整数 kk,表示一个询问元素。

输出格式

共 q 行,每行包含两个整数,表示所求元素的起始位置和终止位置。

如果数组中不存在该元素,则返回 -1 -1。

数据范围

1≤n≤1000001≤n≤100000
1≤q≤100001≤q≤10000
1≤k≤100001≤k≤10000

输入样例:

6 3
1 2 2 3 3 4
3
4
5

输出样例:

3 4
5 5
-1 -1

题解

五点七边的二分法

五点七边 二分法 红蓝区域边界法

l = -1,r = N
while(l+1 != r)
	m=(l+r)/2
	if IsBlue(m)
		l = m;
	else
		r = m
return l or r

第一个细节 l,r 边界问题

为什么 l 边界初始化为 -1?

如果数组从开始就是红色边界,那么就如果 l 边界初始化为0的话,那么就会错误,不会出现蓝色区域

同理为什么 r 边界初始化为 N而不是N-1 ?

整个数组如果从刚开始就是蓝色区域,那么 r边界定义为N-1 会处于蓝色区域内,就会出现错误,不会出现红色区域

第二个细节 m是否始终处于[0,N)以内

在这里插入图片描述

第三个细节,更新指针时,能不能写成l = m+1,r = m-1

在这里插入图片描述

第四个细节 代码会不会陷入死循环

在这里插入图片描述

流程

1. 确定分界线

2.建模,划分红蓝区域,确定IsBlue()

3. 确定返回l 还是 r

4. 套用算法模版

代码

import java.util.Scanner;

public class day0705_0 {
    
    
    public static void main(String[] args) {
    
    
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int q = sc.nextInt();
        int arr[] = new int[n];
        for(int i=0;i<n;i++){
    
    
            arr[i] = sc.nextInt();
        }
        for(int j =0;j<q;j++){
    
    
            int temp = sc.nextInt();
            int res1 = binary_search1(arr,n,temp);
            int res2 = binary_search2(arr,n,temp);
            System.out.println(res1+" "+res2);
        }
    }
    public static boolean isBlue1(int num,int x){
    
    
        if(num < x ) return true;
        else return false;
    }
    public static int binary_search1(int [] arr,int len,int x){
    
    
        int l = -1,r = len;
        while(l+1<r){
    
    
            int mid = (l+r)>>1;
            if(isBlue1(arr[mid],x)){
    
    
                l = mid;
            }else{
    
    
                r = mid;
            }
        }
        if(r == len){
    
    
            return -1;
        }
        if(arr[r] == x) return r;
        else return -1;
    }
    public static int binary_search2(int []arr,int len,int x){
    
    
        int l =-1,r = len;
        while(l+1 <r){
    
    
            int mid = (l+r) >>1;
            if(isBlue2(arr[mid],x)){
    
    
                l = mid;
            }else{
    
    
                r = mid;
            }
        }
        if(l == -1){
    
    
            return -1;
        }
        if(arr[l] == x) return l;
        else return -1;
    }
    public static boolean isBlue2(int num,int x){
    
    
        if(num<=x) return true;
        else return false;
    }
}

优化后:

import java.util.Scanner;
public class day0705_0 {
    
    
    static int N = 100010;
    public static void main(String[] args) {
    
    
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int q = sc.nextInt();
        int arr[] = new int[N];
        for(int i=0;i<n;i++){
    
    
            arr[i] = sc.nextInt();
        }
        for(int j =0;j<q;j++){
    
    
            int temp = sc.nextInt();
            int res1 = binary_search1(arr,n,temp);
            if(res1 == -1){
    
    
                System.out.println(-1+" "+-1);
                continue;
            }
            int res2 = binary_search2(arr,n,temp);
            System.out.println(res1+" "+res2);
        }
    }
    public static int binary_search1(int [] arr,int len,int x){
    
    
        int l = -1,r = len;
        while(l+1<r){
    
    
            int mid = (l+r)>>1;
            if(arr[mid] <x){
    
    
                l = mid;
            }else{
    
    
                r = mid;
            }
        }
        if(r == len){
    
    
            return -1;
        }
        if(arr[r] == x) return r;
        else return -1;
    }
    public static int binary_search2(int []arr,int len,int x){
    
    
        int l =-1,r = len;
        while(l+1 <r){
    
    
            int mid = (l+r) >>1;
            if(arr[mid] <=x){
    
    
                l = mid;
            }else{
    
    
                r = mid;
            }
        }
        if(l == -1){
    
    
            return -1;
        }
        if(arr[l] == x) return l;
        else return -1;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_54174102/article/details/131551714