Programmerings uppgift Python!

#!/usr/bin/env python
# -*- coding: utf-8 -*-

def simulate_day(array):
        x = [0] * len(array)
        for i in xrange(len(array)):
                if array[i] == 1:
                        x[0 if i == 0 else i - 1] = 1
                        x[len(array) if i == len(array) else i + 1] = 1
                        x[i] = 1
        return x

total = int(raw_input('Antal tomater ? '))

array = [0] * total

for i in xrange(3):
        pos = int(raw_input('Mogen ? '))
        array[pos] = 1

t = int(raw_input('Tid ? '))

for i in xrange(t):
        array = simulate_day(array)

print 'Antal mogna tomater: %d' % (array.count(1))