#ifndef __UTIL_H__
#define __UTIL_H__

#include <functional>
#include <string>

using namespace std;

enum struct OutputMode { HackerRank, Linux };

OutputMode outputMode = OutputMode::Linux;

function<int(int)> f_isspace = [](int x){ return isspace(x); };

string ltrim(const string &str) {
    string s(str);

    s.erase(
        s.begin(),
        find_if(s.begin(), s.end(), not_fn(f_isspace))
        );

    return s;
}

string rtrim(const string &str) {
    string s(str);

    s.erase(
        find_if(s.rbegin(), s.rend(), not_fn(f_isspace)).base(),
        s.end()
        );

    return s;
}

vector<string> split(const string &str) {
    vector<string> tokens;

    string::size_type start = 0;
    string::size_type end = 0;

    while ((end = str.find(" ", start)) != string::npos) {
        tokens.push_back(str.substr(start, end - start));

        start = end + 1;
    }

    tokens.push_back(str.substr(start));

    return tokens;
}

#endif