Cued In

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>

int main() {
	int n;
	std::cin >> n;
	std::vector<int> balls;
	for (int i = 0; i < n; ++i) {
		std::string str;
		std::cin >> str;
		if (str == "red") {
			balls.push_back(1);
		} else if (str == "yellow") {
			balls.push_back(2);
		} else if (str == "green") {
			balls.push_back(3);
		} else if (str == "brown") {
			balls.push_back(4);
		} else if (str == "blue") {
			balls.push_back(5);
		} else if (str == "pink") {
			balls.push_back(6);
		} else {
			balls.push_back(7);
		}
	}
	std::sort(balls.begin(), balls.end());
	if (n == 0) {
		std::cout << "0\n";
	} else if (balls[balls.size() - 1] == 1) {
		std::cout << "1\n";
	} else {
		int reds = 0;
		int total = 0;
		for (int i = 0; i < n; ++i) {
			if (balls[i] == 1) {
				++reds;
			} else {
				total += balls[i];
			}
		}
		total += reds * (1 + balls[balls.size() - 1]);
		std::cout << total << '\n';
	}
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED

Test 13

Verdict: ACCEPTED

Test 14

Verdict: ACCEPTED

Test 15

Verdict: ACCEPTED

Test 16

Verdict: ACCEPTED

Test 17

Verdict: ACCEPTED

Test 18

Verdict: ACCEPTED