สมมติว่ามีรายการตัวเลข nums สตริง op แทนตัวดำเนินการ "+", "-", "/", "*" และค่า val เราต้องดำเนินการกับทุกตัวเลขใน nums กับ val และคืนผลลัพธ์
เช่น ถ้าอินพุตคือ [5,3,8] op='*', val=3 ผลลัพธ์จะเป็น [15, 9, 24]
ขั้นตอนการแก้ปัญหา
- res := รายการใหม่
- สำหรับแต่ละ i ใน nums
- ถ้า op == '+' ใส่ i+val เข้า res
- ถ้า op == '-' ใส่ i-val เข้า res
- ถ้า op == '*' ใส่ i*val เข้า res
- ถ้า val ไม่ใช่ 0 ใส่ i//val เข้า res
- คืนค่า res
ตัวอย่างโค้ด
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]
