Задача 18: Цветной дождь

Условие задачи:

В Банановой республике очень много холмов, соединенных мостами. На химическом заводе произошла авария, в результате чего испарилось экспериментальное удобрение "зован". На следующий день выпал цветной дождь, причем он прошел только над холмами. В некоторых местах падали красные капли, в некоторых – синие, а в остальных – зеленые, в результате чего холмы стали соответствующего цвета. Президенту Банановой республики это понравилось, но ему захотелось покрасить мосты между вершинами холмов так, чтобы мосты были покрашены в цвет холмов, которые они соединяют. К сожалению, если холмы разного цвета, то покрасить мост таким образом не удастся. Посчитайте количество таких "плохих" мостов.

Входные данные:

В первой строке входных данных содержится число \(N\) \((0 < N ≤ 100)\) – количество холмов. Далее идет матрица смежности, описывающая наличие мостов между холмами (1 – мост есть, 0 – нет). После матрицы смежности идёт пустая строка, и в последней строке записано \(N\) чисел, обозначающих цвет холмов: 1 – красный; 2 – синий; 3 – зеленый.

Выходные данные:

Выведите одно число – количество "плохих" мостов.

Решение:

n = int(input())
u = [[int(i) for i in input().split()] for i in range(n)]

input()
per = [int(i) for i in input().split()]

s = 0
for i in range(n):
    for j in range(n):
        if u[i][j]:
            if per[i] != per[j]:
                s += 1

print(s // 2)

У вас включен блокировщик рекламы. Пожалуйста, выключите его чтобы увидеть решения. На этом сайте не так много рекламы, а кушать что-то надо. Вот котик