leetcode-34-在排序数组中查找元素的第一个和最后一个位置(find first and last position of element in sorted array)-java

题目及测试

package pid034;
/*在排序数组中查找元素的第一个和最后一个位置

给定一个按照升序排列的整数数组 nums，和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

你的算法时间复杂度必须是 O(log n) 级别。

如果数组中不存在目标值，返回 [-1, -1]。

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]


*/
public class main {
	
	public static void main(String[] args) {
		int[][] testTable1=new int[][]{{5,7,7,8,8,10},{5,7,7,8,8,10}};
		int[] testTable2=new int[]{8,6};
			for(int i=0;i<testTable1.length;i++){
				test(testTable1[i],testTable2[i]);
			}
		
	}
		 
	private static void test(int[] ito1, int ito2) {
		Solution solution = new Solution();
		int[] rtn;
		long begin = System.currentTimeMillis();
		
		rtn = solution.searchRange(ito1,ito2);//执行程序
		long end = System.currentTimeMillis();		
		
	    System.out.print(rtn[0]+" "+rtn[1]);
		
		System.out.println();
		System.out.println("耗时：" + (end - begin) + "ms");
		System.out.println("-------------------");
	}

}

解法1（成功，7ms，较快）

使用二分查找的办法，先用二分查找得到target的位置，然后从找到的位置向前和向后寻找相同的target，直到数组边界或者不同，从而得到first和last

如果二分查找找不到（end<begin），则返回-1，-1

package pid034;

import java.util.HashMap;

public class Solution {
public int[] searchRange(int[] nums, int target) {
    int length=nums.length;
    int[] result=new int[]{-1,-1};
    if(length<=0){   	
    	return result;
    }
    int begin=0;
    int end=length-1;
    int mid=begin/2+end/2;    
    while(true){
    	int now=nums[mid];
    	if(now==target){
    		break;
    	}
    	if(now>target){
    		end=mid-1;
    	}
    	else{
    		begin=mid+1;
    	}
    	mid=(begin+end)/2;   
    	if(end<begin){
    		return result;
    	}   	
    }
    int first=mid;
    int last=mid;
    while(first>=0&&nums[first]==target){
    	first--;
    }
    first++;
    while(last<=length-1&&nums[last]==target){
    	last++;
    }
    last--;
    result[0]=first;
    result[1]=last;
    
	
	return result;
    }
}

解法2（成功，0ms，极快）

先用二分法找到第一个target，再以第一个target为begin，继续二分法，找到最后一个target

    public int[] searchRange(int[] nums, int target) {
        int length=nums.length;
        int[] result=new int[]{-1,-1};
        if(length==0){
        	return result;
        }
        // 先找到第一个target
        int begin=0;
        int end=length-1;
        while(begin<=end){
        	int mid=(begin+end)/2;
        	int now=nums[mid];
        	if(now==target){
        		if(mid==0){
        			// 一定是第一个
        			result[0]=mid;
        			break;
        		}
        		if(nums[mid-1]!=target){
        			// mid为target，mid-1不是，说明mid是第一个target
        			result[0]=mid;
        			break;
        		}
        		// 不然，说明mid不是第一个target
        		end=mid-1;
        		continue;
        	}
        	if(now<target){
        		begin=mid+1;
        	}else{
        		end=mid-1;
        	}
        }
        if(result[0]==-1){
        	return result;
        }
        // 根据第一个target，找到最后一个target
        begin=result[0];
        end=length-1;
        while(begin<=end){
        	int mid=(begin+end)/2;
        	int now=nums[mid];
        	if(now==target){
        		if(mid==length-1){
        			// 一定是最后一个
        			result[1]=mid;
        			break;
        		}
        		if(nums[mid+1]!=target){
        			// mid为target，mid+1不是，说明mid是最后一个target
        			result[1]=mid;
        			break;
        		}
        		// 不然，说明mid不是最后一个target
        		begin=mid+1;
        		continue;
        	}
        	// 因为begin从第一个target开始，所以mid一定>=target
        	end=mid-1;
        }
    	
    	return result;
    }