[leetcode]Search in Rotated Sorted Array II

最新推荐文章于 2019-05-08 14:56:52 发布

原创最新推荐文章于 2019-05-08 14:56:52 发布 · 106 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

面试题

leetcode

算法

本文探讨了在含有重复元素的旋转有序数组中查找目标值的问题，介绍了如何通过改进二分搜索算法来解决这一挑战。

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;
	}