计蒜客 挑战难题 元素移除

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

如：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 input = new Scanner(System.in);
        int n = input.nextInt();
        int[]a=new int[n];
        for(int i=0;i<n;i++)
        {
            a[i]=input.nextInt();
        }
        int elem = input.nextInt();
        System.out.println(removeElement(a,n,elem));
    }
    public static int removeElement(int a[],int n,int elem)
    {
        int s=n;
        for(int i=0;i<n;i++)
        {
            if(elem == a[i])
            {
                remove(a,elem);
                s = s - 1;
            }
        }
        return s;
    }
   private static int[] remove(int[] arr, int num) 
   {
        int[] tmp = new int[arr.length - 1];
        int idx = 0;
        boolean hasRemove = false;
        for (int i = 0; i < arr.length; i++) 
        {
            if (!hasRemove && arr[i] == num) 
            {
                hasRemove = true;
                continue;
            }
            tmp[idx++] = arr[i];
        }
        return tmp;
    }
}