C++ Code for The Relationship (TCS Codevita) | PrepInsta
The Relationship
TCS CodeVita is a coding competition organized by TCS every year, in search of world’s best coder. This is a global level coding competition in which coders from all around the world compete for the title of World’s Best Coder. The Relationship is one of the sample problem of this year TCS CodeVita season 11 competition.
Question -: The people of Kingdoms of Westeros are very organized. They are very devoted to their leader and cannot live without a leader even for a single day. Marriages, divorces, births, and deaths are everyday things in Westeros and because of these events people of Westeros need to spend their time on choosing the leader. People badly need a programme to find the new leader in case of these events. The criteria to choose the leader is as below.
C++
C++
Run
#include < iostream>
#include < vector>
#include < map>
#include < algorithm>
struct Person {
std::string firstname;
std::string familyname;
std::string birthfamilyname;
std::string gender;
std::string fathername;
std::string spousename;
int age;
std::string leadername;
};
std::map people;
bool comparePeople(std::string person1, std::string person2) {
return people[person1].firstname < people[person2].firstname;
}
void chooseLeader(std::string familyname) {
std::vector< std::string> familyMembers;
for (const auto& person : people) {
if (person.second.familyname == familyname) {
familyMembers.push_back(person.first);
}
}
std::sort(familyMembers.begin(), familyMembers.end(), comparePeople);
std::string leader = familyMembers[0];
for (const auto& member : familyMembers) {
if (people[member].age > people[leader].age) {
leader = member;
}
}
people[leader].leadername = leader;
for (const auto& member : familyMembers) {
people[member].leadername = leader;
}
}
int main() {
int N, E;
std::cin >> N;
for (int i = 0; i < N; ++i) {
Person person;
std::cin >> person.firstname >> person.familyname >> person.birthfamilyname >> person.gender
>> person.fathername >> person.spousename >> person.age;
person.leadername = person.firstname;
people[person.firstname] = person;
}
std::cin >> E;
std::cin.ignore();
for (int i = 0; i < E; ++i) {
std::string event;
std::getline(std::cin, event);
if (event == "PA") {
std::vector alivePeople;
for (const auto& person : people) {
if (person.second.age > 0) {
alivePeople.push_back(person.first);
}
}
std::sort(alivePeople.begin(), alivePeople.end(), comparePeople);
for (const auto& person : alivePeople) {
std::cout << people[person].firstname << " "
<< people[person].familyname << " "
<< people[person].birthfamilyname << " "
<< people[person].gender << " "
<< people[person].fathername << " "
<< people[person].spousename << " "
<< people[person].age << " "
<< people[person].leadername << std::endl;
}
} else if (event.substr(0, 2) == "PO") {
std::string name = event.substr(3);
std::cout << people[name].firstname << " "
<< people[name].familyname << " "
<< people[name].birthfamilyname << " "
<< people[name].gender << " "
<< people[name].fathername << " "
<< people[name].spousename << " "
<< people[name].age << " "
<< people[name].leadername << std::endl;
} else if (event.substr(0, 2) == "MA") {
std::string person1, person2;
std::cin >> person1 >> person2;
std::string family1 = people[person1].familyname;
std::string family2 = people[person2].familyname;
people[person1].spousename = person2;
people[person2].spousename = person1;
if (family1 != family2) {
chooseLeader(family1);
chooseLeader(family2);
}
} else if (event.substr(0, 2) == "DI") {
std::string person1, person2;
std::cin >> person1 >> person2;
people[person1].spousename = "NA";
people[person2].spousename = "NA";
std::string family1 = people[person1].familyname;
std::string family2 = people[person2].familyname;
chooseLeader(family1);
chooseLeader(family2);
} else if (event.substr(0, 2) == "BI") {
std::string name, gender, father, mother;
std::cin >> name >> gender >> father >> mother;
people[name].familyname = people[father].familyname;
people[name].birthfamilyname = people[father].familyname;
people[name].fathername = father;
people[name].spousename = "NA";
chooseLeader(people[father].familyname);
} else if (event.substr(0, 2) == "DE") {
std::string person;
std::cin >> person;
people[person].age = -1;
std::string family = people[person].familyname;
chooseLeader(family);
} else if (event.substr(0, 2) == "YP") {
int years;
std::cin >> years;
for (auto& person : people) {
if (person.second.age > 0) {
person.second.age += years;
}
}
}
}
return 0;
}
