python元素移除问题

给定一个已经升序排好序的数组，以及一个数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)

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