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 10000int 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;}