计蒜客 挑战难题 寻找插入位置

题目：
给定一个已经升序排好序的数组，以及一个数target，如果target在数组中，返回它在数组中的位置。

否则，返回target插入数组后它应该在的位置。

假设数组中没有重复的数。以下是简单的示例：

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

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

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

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

提示：输入一个整数n，以及其对应的数组A[n]，最后输入target

searchInsert(int A[], int n, int target)

样例1

输入：

3
1 3 5
2

输出：

1

code:

#include 
int searchInsert(int A[],int n,int target)
{
    int i;
    for(i=0;iif(*(A+i) == target)
            break;
        else if(*(A+i)>target)
                break;
    }
    return i;
}
int main()
{
    int N,i,element;
    scanf("%d",&N);
    int num[N];
    for(i=0;i"%d",&num[i]);
    }
    scanf("%d",&element);
    printf("%d\n",searchInsert(num,N,element));
    return 0;   
}