Array 101. Дан массив размера N и целое число K \((1 \leq K \leq N)\). Перед элементом массива с порядковым номером K вставить новый элемент с нулевым значением.

Решение на Python 3

import random

N = random.randrange(2,21)
K = random.randrange(0,N)
a = [random.randrange(1,5) for i in range(N)]

print("N = ", N)
print("K = ", K+1)
print("Array:\n",a)

print("Modified Array 1:\n",a[:K] + [0] + a[K:])

a.append(-999)
for i in range(N,K,-1) :
a[i] = a[i-1]
a[K] = 0

print("Modified Array 2:\n",a)
print("Length:\n",len(a))