Update problem_1006.cpp