До 5 строчки решения достаточно будет. Дальнейшие упрощения не проверял из-за их ненужности.