Задача про игру в хоккей

Решение на С++

#include <bits/stdc++.h>

using namespace std;
int main()
{
system("chcp 1251");
int k,v,v1 = 0,v2 = 0;
cout << "Вводите штрафное время команд в минутах 2 или 5, окончание ввода номер команды 0" << endl;
cout << "Команда 1 или 2, 0-выход"<<endl;
cin >> k;
switch(k){
case 1:{
cout<<"Штраф = ";
cin >> v;
v1 += v;
cout << "Команда 1 = " << v1 << " мин Команда 2 = " << v2 << " мин" << endl;
break;
}
case 2:{
cout << "Штраф = ";
cin >> v;
v2 += v;
cout << "Команда 1 = " << v1 << " мин Команда 2 = " << v2 << " мин" << endl;
}
}

cout<<"Игра окончена" << endl;
return 0;
}

Решение на Python

print("Вводите штрафное время команд в минутах 2 или 5, окончание ввода номер команды 0")
v1=0
v2=0
print("Команда 1 или 2, 0-выход")
k = int(input())
if k == 1:
print("Штраф=")
v = int(input())
v1 += v
print("Команда 1=", v1, " мин Команда = ", v2, " мин")
if k == 2:
print("Штраф=")
v = int(input())
v2 += v
print("Команда 1=", v1, " мин Команда = ", v2, " мин")

print("Игра окончена")