个人任务4.搜索插入位置_搜索插入位置作者: turbo时间限制: 1s章节: 课程设计-CSDN博客

本文链接：https://blog.csdn.net/xiatutut/article/details/125584316

作者: 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;
}