hello-algo/ko/codes/rust/chapter_searching/binary_search_edge.rs

/*
 * File: binary_search_edge.rs
 * Created Time: 2023-08-30
 * Author: night-cruise ([email protected])
 */

mod binary_search_insertion;

use binary_search_insertion::binary_search_insertion;

/* 가장 왼쪽에 있는 target을 이진 탐색 */
fn binary_search_left_edge(nums: &[i32], target: i32) -> i32 {
    // target의 삽입 위치를 찾는 것과 같다
    let i = binary_search_insertion(nums, target);
    // target을 찾지 못했으므로 -1을 반환한다
    if i == nums.len() as i32 || nums[i as usize] != target {
        return -1;
    }
    // target을 찾았으므로 인덱스 i를 반환한다
    i
}

/* 가장 오른쪽에 있는 target을 이진 탐색 */
fn binary_search_right_edge(nums: &[i32], target: i32) -> i32 {
    // 가장 왼쪽에 있는 target + 1을 찾는 문제로 바꾼다
    let i = binary_search_insertion(nums, target + 1);
    // j는 가장 오른쪽의 target을 가리키고, i는 target보다 큰 첫 번째 원소를 가리킨다
    let j = i - 1;
    // target을 찾지 못했으므로 -1을 반환한다
    if j == -1 || nums[j as usize] != target {
        return -1;
    }
    // target을 찾았으므로 인덱스 j를 반환한다
    j
}

/* 실행 코드 */
fn main() {
    // 중복 원소를 포함하는 배열
    let nums = [1, 3, 6, 6, 6, 6, 6, 10, 12, 15];
    println!("\n배열 nums = {:?}", nums);

    // 왼쪽 경계와 오른쪽 경계의 이진 탐색
    for target in [6, 7] {
        let index = binary_search_left_edge(&nums, target);
        println!("가장 왼쪽에 있는 원소 {}의 인덱스: {}", target, index);
        let index = binary_search_right_edge(&nums, target);
        println!("가장 오른쪽에 있는 원소 {}의 인덱스: {}", target, index);
    }
}