Update a1-2_general-case.cpp