[leetcode]Search in Rotated Sorted Array II

Search in Rotated Sorted Array II

Follow up for "Search in Rotated Sorted Array":
What if duplicates are allowed?

Would this affect the run-time complexity? How and why?

Write a function to determine if a given target is in the array.

 Search in Rotated Sorted Array的进化,考虑了有重复数字的情况,在 Search in Rotated Sorted Array中,可以比较容易的判断target与a[mid]之间的位置关系,从而决定该搜索哪半边,但是加入重复数字的情况后,情况变得略微复杂了。

例如[4,4,4,4,4,5,3,4,4],target = 3

第一次得到的a[mid] = 4

如果序列数字都是唯一的,像 Search in Rotated Sorted Array中的情况,我们可以很容易的判断出4在那边,并只需要搜索相应的半边就可以了。

但是考虑了重复,前半段序列和后半段序列出现了交集,我们需要判断一下这个mid是属于前半段的还是后半段的。

值得庆幸的是,target不需要判断,比如target = 4,因为只是为了证明存在性,因此我们可以假设是任意一个4既可以是前半段的4,也可以是后半段的4。

 

public boolean search(int[] a,int target){
		if(a == null || a.length == 0){
			return false;
		}
		int begin = 0;
		int end = a.length -1;
		while(begin <= end){
			int mid = (begin + end)>>1;
			if(a[mid] == target){
				return true;
			}else if(a[mid] < target){
				//mid在后半段,target在前半段时
				if(findPlace(a, begin, end, mid) == -1 && target > a[end]){
						end = mid - 1;
					}else{
						begin = mid + 1;
					}
			}else{
				//mid在前半段,target在后半段时
				if(findPlace(a, begin, end, mid) == 1 && target < a[begin]){
					begin = mid + 1;
				}else{
					end = mid - 1;
				}
			}
		}
		return false;
	}

	private int findPlace(int[] a,int begin,int end,int index){
		//1表示在前半段,-1表示在后半段,0表示順序序列
		if(a[index] < a[begin] ){
			return -1;
		}
		if(a[index] == a[begin]){
			for(int i = index; i >= begin; i--){
				if(a[i] != a[begin]){
					return -1;
				}
			}
			return 1;
		}
		if(a[index] > a[end]){
			return 1;
		}
		if(a[index] == a[end]){
			for(int i = index; i <= end; i++){
				if(a[i] != a[end]){
					return 1;
				}
			}
			return -1;
		}
		return 0;
	}

 

你可能感兴趣的:(LeetCode)