题目
搜索旋转排序数组
假设有一个排序的按未知的旋转轴旋转的数组(比如,0 1 2 4 5 6 7 可能成为4 5 6 7 0 1 2)。给定一个目标值进行搜索,如果在数组中找到目标值返回数组中的索引位置,否则返回-1。
你可以假设数组中不存在重复的元素。
样例
给出[4, 5, 1, 2, 3]和target=1,返回 2
给出[4, 5, 1, 2, 3]和target=0,返回 -1
解题
直接找,时间复杂度O(N)
class Solution: """ @param A : a list of integers @param target : an integer to be searched @return : an integer """ def search(self, A, target): # write your code here if A == None: return -1 l = len(A) for i in range(l): if A[i]==target: return i return -1
这里题目说了是有序的数组进行了旋转,为什么直接遍历这个蠢的方法?已知是排序的,只是进行了旋转,能否用二分法?
public class Solution { /** *@param A : an integer rotated sorted array *@param target : an integer to be searched *return : an integer */ public int search(int[] A, int target) { if (A == null || A.length == 0) { return -1; } int start = 0; int end = A.length - 1; int mid; while (start + 1 < end) { mid = (end + start) / 2; if (A[mid] == target) { return mid; } if (A[start] < A[mid]) { // 左边升序 // situation 1, red line if (A[start] <= target && target <= A[mid]) { end = mid; } else { start = mid; } } else { // 右边升序 // situation 2, green line if (A[mid] <= target && target <= A[end]) { start = mid; } else { end = mid; } } } // while if (A[start] == target) { return start; } if (A[end] == target) { return end; } return -1; } }