1. 程式人生 > >【計蒜客】元素移除

【計蒜客】元素移除

題目描述

給定一個數組和一個數(該數不一定在陣列中),從數組裡刪掉這個數字,返回剩下的陣列長度。
如: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

AC程式碼

import java.util.Scanner;

public class Main {
	public static void main(String[] args){
		Scanner sc = new Scanner(System.in);
		int n = sc.nextInt();
		int[] arr = new int[n];
		for(int i = 0; i < n; i++){
			arr[i] = sc.nextInt();
		}
		int elem = sc.nextInt();
		System.out.println(removeElement(arr, n, elem));
		
	}

	private static int removeElement(int[] arr, int n, int elem) {
		int count = 0;
		for(int i = 0; i < n; i++){
			if(arr[i] == elem){
				count++;
			}
		}
		return n - count;
	}
}