作者: Turbo 时间限制: 1S 章节: 课程设计
问题描述:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例 1:
输入:
4
1 3 5 6
5
输出: 2
示例 2:
输入:
4
1 3 5 6
2
输出: 1
输入说明:
输入三行:
第一行输入一个整数n表示数组nums的长度。
第二行输入n个整数表示数组nums的元素。
第三行输入一个整数表示需要查找的目标值target.
提示:
1 <= n <= 10^4
-10^4 <= nums[i] <= 10^4
nums 为 无重复元素 的 升序 排列数组
-10^4 <= target <= 10^4
输出说明:
输出一个整数表示结果。
输入范例:
4
1 3 5 6
7
输出范例:
4
#include<iostream>
#include<vector>
#include<string>
#include<sstream>
#include<queue>
#include<stack>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<stdlib.h>
#include<stdio.h>
#include<fstream>
#include<iomanip>
#include<utility>
using namespace std;
int a[10001]={0};
int fun(int a[],int n,int m){//用来寻找m是否在数组a中
int i=0,j=n-1;
while(i<j){//使用二分查找降低时间复杂度
int mid=(i+j)/2;
if(m<a[mid])
j=mid;
else if(m>a[mid])
i=mid+1;
else
return mid;//当m等于a[mid]时,返回下标mid
}
return -1;//如果i>j而且没找到相等的值时,则返回-1
}
int weizhi(int a[],int n,int m){//如果没有在a中找到该数,则就寻找应该插入的位置
int i=0,j=n-1;
if(m>a[n-1])//如果m大于最后一个元素,则返回n值
return n;
if(m<a[0])//如果m小于第一个元素,返回0
return 0;
while(i<j){//仍然使用二分查找
int mid=(i+j)/2;
if(m<a[mid])
j=mid;
else if(m>a[mid])
i=mid+1;
if(i+1==j)
break;
}
return i;
}
int main(){
int n,m;
cin>>n;
for(int i=0;i<n;++i)
cin>>a[i];
cin>>m;
int x=fun(a,n,m);
if(x==-1)//如果没有找到给数,则就调用weizhi函数查找其位置
x=weizhi(a,n,m);
cout<<x;
return 0;
}
这个题在LeetCode是有原题的,但是OJ上的题目要求应该是和力扣上的不一样,我的代码放在力扣上可能会有问题,但是在OJ里面是AC的。
原题链接: 35. 搜索插入位置