traffic_replay: Improve assign_groups() performance with large domains