q4.cpp

Go to the documentation of this file.

/* perform breakup of k unhappiest couples, and assign the
   girls who became single, another (new) boyfriend
 */

#include "../boys/allboys.h"
#include "../girls/allgirls.h"
#include "../gifts/gift.h"
#include "../couples/couple.h"
#include "../utils/read_data.h"
#include "../utils/couple_utils.h"
#include "../utils/logger/logger.h"

#include "../configs.h"

int main(int argc, char **argv)
{
    int k = K;
    if (argc >= 2)
        k = atoi(argv[1]);
    Logger logger(LOG_FILE);

    std::vector<Boy*>  boys  = readAllBoys(BOY_FILE);
    std::vector<Girl*> girls = readAllGirls(GIRL_FILE);
    std::vector<Gift> gifts = readAllGifts(GIFT_FILE);

    logger.info("Processing for question 4", true);
    logger.log("boys", std::to_string(boys.size()) + " record read", true);
    logger.log("girls", std::to_string(girls.size()) + " record read", true);
    logger.log("gifts", std::to_string(gifts.size()) + " record read", true);

    std::vector<Couple> couples = makeCouples(boys, girls, &logger);

    logger.log("couples", std::to_string(couples.size()) + " formed", true);

    logger.info("Couples list", true);
    for (int i = 0; i < (int)couples.size(); i++)
        logger.log("couple"+std::to_string(i+1),
                   couples[i].boy->name+" and "+couples[i].girl->name, true);

    performGifting(&couples, &gifts, &logger);

    if (k > (int)couples.size()) {
        logger.log("Error", "K is more than couples", true);
        return 1;
    }

    std::vector<Couple> hc = getKUnhappiestCouples(couples, k);
    logger.info(std::to_string(k)+ " unhappiest Couples list", true);
    for (int i = 0; i < (int)hc.size(); i++)
        logger.log("couple"+std::to_string(i+1),
                   hc[i].boy->name+" and "+hc[i].girl->name, true);

    logger.info("Performing breakup of k unhappiest couples and assign the girls new boyfriends", true);
    performBreakupAndPairAgain(hc, &couples, &boys, &girls, &logger);
    return 0;
}