1. 程式人生 > >計蒜客挑戰難題:元素移除

計蒜客挑戰難題:元素移除

給定一個數組和一個數(該數不一定在陣列中),從數組裡刪掉這個數字,返回剩下的陣列長度。

如: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 elem = sc.nextInt(); System.out.println(removeElement(A,n,elem)); } public
static int removeElement(int[] A, int n, int elem) { int length = 0; for(int i = 0; i < n; i++){ if(elem != A[i]) ++length; } return length; } }