A gunman suspected of opening fire on a group of young swimmers on a riverbank, leaving three of them dead, has been arrested.