Вам нужно добавить shouldReset
var и сделать это внутри cellForRowAt
cell.backgroundColor = shouldReset ? UIColor.red : UIColor.blue
всякий раз, когда вам нужно изменить цвет, делайте
collectionView.reloadData()
и управлять
shouldReset
будет true/false