utils.hpp

#ifndef UTILS_H_
#define UTILS_H_

#include <filesystem>
#include <map>
#include <vector>
#include <iostream>

enum PrintFormat { Prompt, Solution };

// map for keeping the names of directories
// promptInputDirectory - directory for in files formatted for prompt
// promptInputDirectory - directory for in files formatted for the solution
const std::map<std::string, std::string> dirs = {{"promptInputDirectory", "in"},
                                                 {"solutionInputDirectory", "solution-in"}};

template <typename F, typename S>
std::ostream &operator<<(std::ostream &os, const std::pair<F, S> &p) {
    return os << "(" << p.first << ", " << p.second << ")";
}

template <typename T>
std::ostream &operator<<(std::ostream &os, const std::vector<T> &v) {
    os << "{";
    typename std::vector<T>::const_iterator it;
    for (it = v.begin(); it != v.end(); ++it) {
        if (it != v.begin()) {
            os << ", ";
        }
        os << *it;
    }
    return os << "}";
}

void println() { std::cout << '\n'; }
template <typename Head, typename... Tail>
void println(Head H, Tail... T) {
    std::cout << ' ' << H;
    println(T...);
}

void setupDirectories() {
    for (const auto &dir : dirs) {
        if (!std::filesystem::create_directory(dir.second)) {
            // Only warning, because you can fail to create a directory if it exists.
            std::cerr << "Warning: Could not create directory " << dir << std::endl;
        }
    }
}

template <typename T>
concept ConvertibleToInt64_t = std::convertible_to<T, int64_t>;

template<ConvertibleToInt64_t T>
int64_t changeVectorToInt64_t(std::vector<T> &v) {
    int64_t result = 0;
    for (int64_t i = 0; i < v.size(); i++) {
        result += (i + 1) * v[i];
    }
    return result;
}

template<ConvertibleToInt64_t T>
int64_t changeVectorOfPairsToInt64_t(std::vector<std::pair<T, T>> &v) {
    int64_t result = 0;
    for (int64_t i = 0; i < v.size(); i++) {
        result += (2*i + 1) * v[i].first + (2*i + 2) * v[i].second;
    }
    return result;
}

#endif