Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

Python에서 요소 제거


배열 num과 다른 값 val이 있다고 가정하면 해당 값의 모든 인스턴스를 제자리에서 제거하고 새 길이를 찾아야 합니다.

따라서 입력이 [0,1,5,5,3,0,4,5] 5와 같으면 출력은 5가 됩니다.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

  • 개수 :=0

  • 숫자의 각 인덱스 i에 대해

    • nums[i]가 val과 같지 않으면 -

      • nums[count] :=nums[i]

    • 개수 :=개수 + 1

  • 반환 횟수

예시

더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −

class Solution:
   def removeElement(self, nums, val):
      count = 0
      for i in range(len(nums)):
         if nums[i] != val :
            nums[count] = nums[i]
            count +=1
      return count
ob = Solution()
print(ob.removeElement([0,1,5,5,3,0,4,5], 5))

입력

[0,1,5,5,3,0,4,5], 5

출력

5