diff --git a/exercises/ex2.cpp b/exercises/ex2.cpp index 9e8cd8e..3873827 100644 --- a/exercises/ex2.cpp +++ b/exercises/ex2.cpp @@ -5,47 +5,22 @@ */ #include +#include + using namespace std; int main() { - int week; + unsigned int week; cout << "Enter week number(1-7): " << endl; cin >> week; - if (week == 1) - { - cout << "Monday" << endl; - } - else if (week == 2) - { - cout << "Tuesday" << endl; - } - else if (week == 3) - { - cout << "Wednesday" << endl; - } - else if (week == 4) - { - cout << "Thursday" << endl; - } - else if (week == 5) - { - cout << "Friday" << endl; - } - else if (week == 6) - { - cout << "Saturday" << endl; - } - else if (week == 7) - { - cout << "Sunday" << endl; - } - else - { - cout << "Invalid input! Please enter week number between 1-7." << endl; - } - - return 0; + string WeekDays[8] = {"Monday", "Tuesday", "Wednesday", "Thursday", + "Friday", "Saturday", "Sunday", "Invalid input! Please enter week number between 1-7."}; + + unsigned int choice = min(week - 1, 7u); + cout << WeekDays[choice] << endl; + + return EXIT_SUCCESS; } \ No newline at end of file