Update solution_0901.cpp