Java - 计蒜客 - 元素移除

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

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

亲爱的小伙伴们，题目是不是很简单呢？

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

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

格式：

输入一个数n，继而输入一个数组A[n],接着输入要删除的元素elem，返回剩余数组长度index.

import java.util.Scanner;  
public class Main{  
	public static void main(String[] args){  
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int[]a = new int[n];
		for(int i=0 ;i<n ;i++){
			a[i] = sc.nextInt();
		}
		int b = sc.nextInt();
		System.out.print(removeElement(a,n,b));
	}
	public static 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;
	}
}