, поэтому я пытаюсь решить проблему с Leetcode.
вход состоит из двух отсортированных списков, а на выходе должно быть их медиана
class Solution(object):
def findMedianSortedArrays(self, nums1, nums2):
"""
:type nums1: List[int]
:type nums2: List[int]
:rtype: float
"""
newlist = nums1 + nums2
newlist = sorted(newlist)
if len(newlist) < 2:
raise ValueError
else:
if len(newlist) % 2 == 0:
return ((newlist[int((len(newlist)//2))-1]) + (newlist[int((len(newlist))//2)]))/2
else:
return newlist[(int((len(newlist)+1)//2))-1]
# nums1 = [1,2]
# nums2 [3,4]
# newlist = [1,2,3,4]
# len(newlist) % 2 = 0
# return should give (2+3)/2 == 2.5000
# return actually give 2.0000
это код, который я пытался сделать путь к спискам в комментариях, чтобы увидеть, что я делаю что-то не так.
вывод должен быть 2.5000 (что я получаю на ноутбуке Jupyter при запуске того же кода), но при запуске на Leetcode я получаю 2.0000 в качестве вывода.
Что-то не так с моим кодом? и как я могу это исправить?
Большое спасибо за помощь