TZI/Лекции/test.py

# https://contest.yandex.ru/contest/78439/problems/A/
"""
В опенспейсе работает a интровертов и b экстравертов. Каждый день они приходят в случайном порядке на работу, входя в опенспейс по одному. Когда интроверт входит в опенспейс, то каждый экстраверт, который уже находится в опенспейсе, говорит ему «Привет!», и интроверт отвечает ему «Привет!». Когда в опенспейс приходит экстраверт, то он говорит каждому сотруднику «Привет!», и каждый сотрудник отвечает ему «Привет!».

Посчитайте, сколько раз было произнесено слово «Привет!» за день.

"""

import random
from tqdm import tqdm

a = int(input())
b = int(input())

privet_n = 0
# количество интровертов в опенспейсе
a_privet = 0
# количество экстравертов в опенспейсе
b_privet = 0

# порядок захода в опенспейс
a_list = [0] * a
b_list = [1] * b
summ_list = a_list + b_list
random.shuffle(summ_list)

for i in tqdm(summ_list):
    # кто заходит - интроверт или экстраверт
    # не все интроверты в опенспейсе
    if i == 0:
        privet_n = privet_n + b_privet * 2
        a_privet += 1
        # print(f"зашел интроверт, приветов: {privet_n}")
    else:
        privet_n = privet_n + (a_privet + b_privet) * 2
        b_privet += 1
        # print(f"зашел экстраверт, приветов: {privet_n}")

print(privet_n)