More efficient set_partition for C++ benchmark

[stephenberry]

Avoids some excess allocations. Still much slower than it could be, but this is trivial cleanup.

[arshajii]

Can merge this after clang-format.

[stephenberry]

Ran clang format on the code