计蒜客 第九题 元素移除

2018-08-13  本文已影响0人  Sourcils

给定一个数组和一个数(该数不一定在数组中),从数组里删掉这个数字,返回剩下的数组长度。

如:A[] = {1, 2, 3, 4, 5},要删除数字 3,那么返回数组长度为 4。

亲爱的小伙伴们,题目是不是很简单呢?

提示:int removeElement(int A[], int n, int elem)

其中,n代表数组长度,elem代表要删掉的元素。

输入格式

第一行输入一个数 n(1≤n≤100),接下来一行 n 个整数,表示数组 A 的所有元素 Ai (0 ≤ Ai ≤100),接着输入要删除的元素 elem(0≤elem≤100)。

输出格式

输出一个整数,表示剩余数组长度。

样例输入

2
3 3
3
样例输出

0

n = int(input())
A = input().split()
elem = int(input())
j = 0 # 记录删除次数
for i in range(n): #删除elem
    if int(A[i-j]) == elem:
        del A[i-j]
        j = j + 1
print(len(A)) # 输出结果
上一篇 下一篇

猜你喜欢

热点阅读