Notes

#include <iostream>
#include <string>
#include <random>
#include <vector>

// Enum representing the different powers that a troop can have
enum class Power {
STRENGTH = 10,
WISDOM = 5,
MAGIC = 3,
NOTHING = 1
};

// Struct representing a single troop
struct Troop {
Power power;
bool alive = true;
};

// Class representing a country
class Country {
public:
Country(const std::string& name) : m_name(name) {}
std::string m_name;
std::vector<Troop> m_troops;

void addTroops(int count) {
std::random_device rd;
std::mt19937 gen(rd());
std::uniform_int_distribution<> dis(1, 4);

for (int i = 0; i < count; i++) {
Troop t;
switch (dis(gen)) {
case 1:
t.power = Power::STRENGTH;
break;
case 2:
t.power = Power::WISDOM;
break;
case 3:
t.power = Power::MAGIC;
break;
case 4:
t.power = Power::NOTHING;
break;
}
m_troops.push_back(t);
}
}

void displayTroops() const {
std::cout << "Troops of " << m_name << ":" << std::endl;
for (const auto& t : m_troops) {
std::cout << " - Power: " << static_cast<int>(t.power) << std::endl;
}
}

void fight(Country& other) {
std::cout << "Fighting between " << m_name << " and " << other.m_name << ":" << std::endl;

for (int i = 0; i < m_troops.size(); i++) {
std::cout << "X";
if (m_troops[i].alive) {
if (other.m_troops[i].alive) {
if (m_troops[i].power > other.m_troops[i].power) {
other.m_troops[i].alive = false;
} else {
m_troops[i].alive = false;
}
}
}
}
std::cout << std::endl;
}

bool hasWon() const {
for (const auto& t : m_troops) {
if (t.alive) {
return true;
}
}
return false;
}

std::string name() const { return m_name; }
};

int main() {
Country wath("Wath");
Country dath("Dath");

// Add a random number of troops to each country
wath.addTroops(10);
dath.addTroops(10);

// Display the number of troops in each country
std::cout << "Troops of Wath: " << wath.name() << " - " << wath.m_troops.size() << std::endl;
std::cout << "Troops of Dath: " << dath.name() << " - " << dath.m_troops.size() << std::endl;

// Display the powers of the troops in each country
wath.displayTroops();
dath.displayTroops();

// Fight between the two countries
wath.fight(dath);

// Determine the winner
if (wath.hasWon()) {
std::cout << "Winner: " << wath.name() << std::endl;
} else {
std::cout << "Winner: " << dath.name() << std::endl;
}

return 0;
}