搜索插入位置

描述

給定一個排序數組和一個目標值，如果在數組中找到目標值則返回索引。如果沒有，返回到它將會被按順序插入的位置。

你可以假設在數組中無重複元素

樣例

[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();
    }
};