Как избавиться от эффекта памяти в React?
Ключи добавьте.

Да, в документации говорится про ключи в связи со списками, но реально область их применения шире. Если хотите, чтобы какой-то элемент пересоздавался в случаях вроде вашего, надо назначить различные ключи элементам, создающимся в различных ветках switch'а, условного оператора и т.п. Например: ключей нет - ключи есть.

Но вообще, правильно будет назначать классы в зависимости от значения какого-то свойства в стейте. Тогда и подобных проблем не будет, ключи не понадобятся.
Крайне рекомендую не пилить свои алгоритмы обработки формы, а использовать готовые проверенные решения вроде react-final-form или formik. Это очень сильно упростит Вам жизнь и избавит от тонн костылей и велосипедов.