Anyone has a better solution than the naive getting subset solution?