Computer >> Máy Tính >  >> Lập trình >> Python

Chương trình tìm danh sách sao chép bằng cách sao chép từng phần tử n lần

Giả sử chúng ta có một danh sách gồm n phần tử; chúng ta phải lặp lại từng phần tử trong danh sách n số lần.

Vì vậy, nếu đầu vào là nums =[1,5,8,3], thì đầu ra sẽ là [1, 1, 1, 1, 5, 5, 5, 5, 8, 8, 8, 8, 3 , 3, 3, 3]

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

  • n:=kích thước của nums
  • ret:=một danh sách mới
  • đối với mỗi num trong nums, thực hiện
    • ret:=ret nối một danh sách với n số num
  • trả lời lại

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn

def solve(nums):
   n = len(nums)
   ret = []
   for num in nums:
      ret += [num]*n
   return ret

nums = [1,5,8,3]
print(solve(nums))

Đầu vào

[1,5,8,3]

Đầu ra

[1, 1, 1, 1, 5, 5, 5, 5, 8, 8, 8, 8, 3, 3, 3, 3]