python挿入位置を探します

昇順に並べられた配列と、配列内にある場合、配列内の位置を返す数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)を入力する

#include<stdio.h>
#define MAX 10000
int searchInsert(int A[], int n, int target);
int main(){
    int n, arr[MAX], target;
    scanf("%d",&n);
    for(int i = 0; i < n; i++){
        scanf("%d",&arr[i]);
    }
    scanf("%d",&target);
    printf("%d
",searchInsert(arr, n, target));
    return 0;
}
int searchInsert(int A[], int n, int target){
    int i = 0;
    for(; i < n; i++){
        if(A[i] >= target)
            return i;
    }
    return i;
}