diff --git a/main.cpp b/main.cpp index a28a84d..e9799ef 100644 --- a/main.cpp +++ b/main.cpp @@ -14,28 +14,37 @@ struct Input { }; Input -input_data(istream& in) { +input_data(istream& in, bool prompt) { size_t number_count; - in >> number_count; - Input inn; - inn.numbers.resize(number_count); - for (size_t i = 0; i < number_count; i++) { - in >> inn.numbers[i]; + if (prompt == true){ + cerr << "input number count"; + in >> number_count; + Input inn; + inn.numbers.resize(number_count); + for (size_t i = 0; i < number_count; i++) { + in >> inn.numbers[i]; + } + size_t bin_count; + in >> inn.bin_count; + return inn; + } + else{ + cout << "input number count"; + in >> number_count; + Input inn; + inn.numbers.resize(number_count); + for (size_t i = 0; i < number_count; i++) { + in >> inn.numbers[i]; + } + size_t bin_count; + in >> inn.bin_count; + return inn; } - size_t bin_count; - in >> inn.bin_count; - return inn; } - - - - - - int main() { - auto in = input_data(cin); + auto in = input_data(cin, true); auto bins = make_histogram(in.numbers, in.bin_count); show_histogram_svg(bins); return 0;