【lintcode笔记】搜索插入位置

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

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

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

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

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

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

//复杂度O(n)
#include <stdio.h>
#include <vector>
using namespace std;
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
        int n = A.size();
        if(n == 0) return 0;
        int i;
        for(i=0;i<=n-1;i++){
            if(A[i] == target){
                return i;
            }else if(A[i] < target){
                if(i+1 == n) return i+1;
            }else{
                return i;
            }
        }
    }
};