코딩 테스트/이진 탐색 문제
이진탐색 문제 - 원형 정렬 상태의 배열에서 이진탐색
꾸준함의 미더덕
2024. 6. 5. 21:17
원형 정렬 상태 배열에서의 이진 탐색
nums 배열에 원형 상태로 데이터가 정렬되어 있을 때, 이진 탐색으로 데이터를 찾는 프로그램을 작성하세요.
O(log n) 유지
배열을 재 정렬하지 않고 풀기
입출력 예시
arr: 4, 5, 6, 7, 8, 0, 1, 2
target: 0
출력: 5
arr: 8, 0, 1, 2, 4, 5, 6, 7,
target: 3
출력: -1
public class Main {
public static int solution(int[] arr, int target) {
return binarySearch(arr, target, 0, arr.length - 1);
}
public static int binarySearch(int[] arr, int target, int left, int right) {
if (left > right) {
return -1;
}
int mid = (left + right) / 2;
if (arr[mid] == target) {
return mid;
}
if (arr[left] <= arr[mid]) { // mid 이하가 정렬되어있을 때 ex) 3 4 5 6 7 8 9 1 2
if (arr[left] <= target && target < arr[mid]) { // 타겟이 arr[left]보다는 크거나 같고 정렬되어 있는 부분에 target이 있는가?
return binarySearch(arr, target, left, mid - 1);
} else {
return binarySearch(arr, target, mid + 1, right);
}
} else { //mid 이상이 정렬되어 있을 때 ex) 8 9 1 2 3 4 5 6 7
if (target <= arr[right] && arr[mid] < target) { // 타겟이 arr[right] 보다는 작거나 같고 정렬되어 있는 부분에 target이 있는가?
return binarySearch(arr, target, mid + 1, right);
} else {
return binarySearch(arr, target, left, mid - 1);
}
}
}
public static void main(String[] args) {
// Test code
int[] nums = {4, 5, 6, 7, 8, 0, 1, 2};
System.out.println(solution(nums, 0)); // 5
System.out.println(solution(nums, 3)); // -1
}
}