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

파이썬에서 리스트의 각 요소와 주어진 값으로 주어진 연산을 수행하는 프로그램

<시간/>

nums라는 숫자 목록이 있고 "+", "-", "/" 또는 "*"와 같은 연산자를 나타내는 또 다른 문자열 op가 있고 또 다른 값 val도 제공된다고 가정합니다. val을 사용하여 숫자를 입력하고 결과를 반환합니다.

따라서 입력이 [5,3,8]과 같으면 출력은 [15, 9, 24]

가 됩니다.

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

  • res:=새 목록
  • 숫자 단위의 각 i에 대해 다음을 수행합니다.
    • op가 '+'와 같으면
      • res 끝에 i+val 삽입
    • 그렇지 않으면 op가 '-'와 같을 때 then
      • res 끝에 i-val 삽입
    • 그렇지 않고 op가 '*'와 같을 때 then
      • res 끝에 i*val 삽입
    • 그렇지 않으면 val이 0이 아닌 경우
      • res 끝에 i/val의 몫 삽입
  • 반환 결과

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

예시

class Solution:
   def solve(self, nums, op, val):
      res=[]
      for i in nums:
         if op=='+':
            res.append(i+val)
         elif op=='-':
            res.append(i-val)
         elif op=='*':
            res.append(i*val)
         elif val:
            res.append(i//val)
      return res
ob = Solution()
nums = [5,3,8]
print(ob.solve(nums, '*', 3))

입력

[5,3,8]

출력

[15, 9, 24]