搜索插入位置

描述

给定一个排序数组和一个目标值，如果在数组中找到目标值则返回索引。如果没有，返回到它将会被按顺序插入的位置。

你可以假设在数组中无重复元素

样例

[1,3,5,6]，5 → 2

[1,3,5,6]，2 → 1

[1,3,5,6]，7 → 4

[1,3,5,6]，0 → 0

思考

1. 如果数组中存在元素i , 使得 A[i] >= target , 则返回元素索引
2. 如果不存在此元素，说明target > A[i], 返回 A.size()

代码

//  By Lentitude


class Solution {
    /** 
     * param A : an integer sorted array
     * param target :  an integer to be inserted
     * return : an integer
     */
public:
    int searchInsert(vector<int> &A, int target) {
        // write your code here
        if (A.empty()){
            return 0;
        }

        //因不清楚原数组是升序,降序,所以先进行升序处理
        sort(A.begin(), A.end());       

        for (int i = 0; i != A.size(); ++i){
            //如果在数组中找到目标值则返回索引.如果没有, 返回到它将会被按顺序插入的位置
            if (A[i] >= target){
                return i;
            }
        }

        //说明 target > A[A.size() - 1]
        return A.size();
    }
};