sh1.cpp - master - nanezz/sh

#include <iostream>
#include <vector>
#include <sstream>
#include <string>

template <typename T>
T square(T x)
{
	return x * x;
}

template <typename T>
std::vector<T> square(const std::vector<T>& v) 
{
    std::vector<T> result;
    result.reserve(v.size());
    for (const T& elem : v) 
    {
        result.push_back(elem * elem);
    }
    return result;
}

int main()
{
    std::cout << "[IN]: ";
    std::string line;
    std::getline(std::cin, line);  

    std::vector<long long> v;
    std::stringstream ss(line);
    long long x;
    char comma;

    while (ss >> x)
    {
        v.push_back(x);
        if (ss.peek() == ',') 
        {
            ss.ignore();
        }
    }

    auto sq = square(v);

    std::cout << "[OUT]: ";
    for (size_t i = 0; i < sq.size(); ++i) 
    {
        if (i > 0) std::cout << ", ";
        std::cout << sq[i];
    }
    std::cout << '\n';

	return EXIT_SUCCESS;
}