Chương trình thực hiện phép toán trên mỗi phần tử danh sách với giá trị cho trước trong Python

Cho danh sách nums, toán tử op ('+', '-', '*', '/') và giá trị val. Thực hiện phép toán trên mỗi phần tử với val và trả về danh sách kết quả.

Ví dụ

Đầu vào: nums=[5,3,8], op='*', val=3.
Đầu ra: [15, 9, 24].

Thuật toán

1. Khởi tạo res = [].
2. Duyệt từng i trong nums:
   • Áp dụng phép toán tương ứng, thêm vào res.
3. Trả về res.

Mã nguồn Python

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))

Đầu vào / Đầu ra

Input: [5,3,8]
Output: [15, 9, 24]