Рассмотрите программный код. Выполните оптимизацию

Var a: array of integer; // Объявление массива целых чисел
  i, n: integer; // объявление целочисленных переменных
begin
  // Запросим с клавиатуры размер массива
  // и запишем его в переменную n
  write('введите n:'); 
  readln(n);
  
  // Теперь выделим память под n элементов массива
  SetLength(a, n);
  
  // теперь запросим с клавиатуры n чисел для массива
  for i := 0 to n-1 do begin
    write('a[', i, ']=');
    readln(a[i]);
  end;
  
  // теперь в цикле уменьшим каждый элемент на единицу
  for i := 0 to n-1 do 
    a[i] := a[i] - 1;
  
  // распечатаем массив
  for i := 0 to n-1 do 
    write(a[i], ' ');
end.

# https://pastebin.com/V2MMpJeN

#

# A.

def function(digit): return str(digit).endswith('0') or str(digit).endswith("1")

print(f"A. function(1230) returns {function(1230)}")

# B.

def function(digit): return digit >= 0 and digit <= 2**8

print(f"B. function(751) returns {function(751)}")

# C.

def function(digit):

if digit < 2: return False

if digit == 2: return True

i=2

lim=int(digit**(1/2))

for n in range(i, lim + 1):

if digit % n == 0: return False

# конец двойной табуляции

return True

# конец табуляции

print(f"C. function(17) returns {function(17)}")

n=int(input("<-- enter digit: "))

print(f"C. function({n}) returns {function(n)}")

print("Число", n, "простое" if function(n) else "составное")