Задача о раскраске поверхности и подсчета площади, занимаемой каждым цветом

Пытаюсь решить задачу, но не получается.

Задача:

Кондитер украшает огромный торт, покрывая прямоугольную поверхность разноцветной глазурью. Для приготовления глазури он смешивает сахарную пудру с лимонным и черничным соком в разных пропорциях, чтобы получить три оттенка синего цвета: светлый, средний и насыщенный. Эти цвета обозначаются числами: 0 для светлого, 1 для среднего и 2 для насыщенного синего.

Чтобы получить красивый узор, он разделяет поверхность торта на вертикальные полосы шириной A1, A2, ..., An сантиметров и горизонтальные полосы высоты B1, B2, ..., Bn сантиметров. В результате полосы делят поверхность торта на n × n прямоугольников. Пересечение вертикальной полосы i и горизонтальной полосы j имеет цветовой номер (i + j) mod 3 для всех 1 ≤ i, j ≤ n.

Меня сбивает с толку формулировка "он разделяет поверхность торта на вертикальные полосы шириной A1, A2, ..., An сантиметров и горизонтальные полосы высоты B1, B2, ..., Bn сантиметров", либо ширину самих полос отобразить, как на первом рисунке, либо ширину и высоту прямоугольников изобразить, как на втором рисунке. Также программа неправильно работает. Что я мог не так сделать?

Рисунок 1

Рисунок 2

Мой код:

def get_pattern(n, a, b):
  colors = [0, 1, 2]
  for j in range(len(b)):
    for i in range(1, len(a)):
      if i == 1:
        if (i + j) % 3 == 0:
          print(colors[((i + j) % 3) - 1], end = ' ')
        elif (i + j) % 3 == 1:
          print(colors[((i + j) % 3) - 1], end = ' ')
        elif (i + j) % 3 == 2:
          print(colors[((i + j) % 3) - 1], end = ' ')
      print((i + j) % 3, end = ' ')
    print()
get_pattern(int(input('Введите n: ')), list(map(int, input('A: ').split())), list(map(int, input('B: ').split())))

# Неправильно выводит цвета
# Введите n: 5
# A: 1 1 1 1 1
# B: 1 1 1 1 1
# 0 1 2 0 1 
# 1 2 0 1 2 
# 1 0 1 2 0 
# 0 1 2 0 1 
# 1 2 0 1 2 

# Должно быть
# Введите n: 5
# A: 1 1 1 1 1
# B: 1 1 1 1 1
# 0 1 2 0 1 
# 2 0 1 2 0
# 1 2 0 1 2
# 0 1 2 0 1
# 2 0 1 2 0

Дополнение от автора, данное в комментариях под первым ответом. С ним становится ясно, для чего даётся ширина полос:

мне такое задание в институте дали, текст задания, в интернете нашёл такую же задачу, и оказалось, что в тексте задания, которую нам дали в институте, отсутствует ещё одно предложение: Помогите кондитеру определить площадь каждого из трех цветов в квадратных сантиметрах. Программа получает на вход целое положительное число n, и две строки, в которых перечисляется ширина и высота полос.