Имеется список целых чисел Измените исходный список, вставив элемент со значением k между всеми соседними элементами списка, имеющими одинаковый знак
Имеется список целых чисел. Измените исходный список, вставив элемент со значением k между всеми соседними элементами исходного списка, имеющими одинаковый знак. Считать, что 0 имеет общий со своими соседями знак.
Формат входных данных
В первой строке вводится последовательность целых чисел через пробел – элементы списка. Во второй строке вводится целое число k.
Формат выходных данных
Требуется вывести измененный список.
входные данные
1 0 9 4 2 9 55
выходные данные
1 55 0 55 9 55 4 55 2 55 9
входные данные
1 -10 -10 3 -5 0 -1 5 9 10 55
выходные данные
1 -10 55 -10 3 -5 55 0
Мой код:
s=list(map(int,input().split()))
k=int(input())
i=0
while i<len(s):
if s[i]>0 and s[i+1]>0:
s[i+2]=s[i+1]
s[i+1]=k
elif s[i]<0 and s[i+1]<0:
s[i+2]=s[i+1]
s[i+1]=k
else:
i+=1
print(*s)