[HackerRank] Compare the Triplets

출처 : https://www.hackerrank.com/challenges/compare-the-triplets/problem

#include <bits/stdc++.h>

using namespace std;

string ltrim(const string &);

string rtrim(const string &);

vector<string> split(const string &);

vector<int> compareTriplets(vector<int> a, vector<int> b) {

    vector<int> v(2);

    for(int i=0; i<3; i++) {

        if(a[i]>b[i]) v[0]++;

        else if(a[i]<b[i]) v[1]++;

    }

    return v;

}

int main()

{

    ofstream fout(getenv("OUTPUT_PATH"));

    string a_temp_temp;

    getline(cin, a_temp_temp);

    vector<string> a_temp = split(rtrim(a_temp_temp));

    vector<int> a(3);

    for (int i = 0; i < 3; i++) {

        int a_item = stoi(a_temp[i]);

        a[i] = a_item;

    }

    string b_temp_temp;

    getline(cin, b_temp_temp);

    vector<string> b_temp = split(rtrim(b_temp_temp));

    vector<int> b(3);

    for (int i = 0; i < 3; i++) {

        int b_item = stoi(b_temp[i]);

        b[i] = b_item;

    }

    vector<int> result = compareTriplets(a, b);

    for (int i = 0; i < result.size(); i++) {

        fout << result[i];

        if (i != result.size() - 1) {

            fout << " ";

        }

    }

    fout << "\n";

    fout.close();

    return 0;

}

string ltrim(const string &str) {

    string s(str);

    s.erase(

        s.begin(),

        find_if(s.begin(), s.end(), not1(ptr_fun<int, int>(isspace)))

    );

    return s;

}

string rtrim(const string &str) {

    string s(str);

    s.erase(

        find_if(s.rbegin(), s.rend(), not1(ptr_fun<int, int>(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;

}