Find heavy ball - level: master

• Time : $O(1)$
• Space : $O(1)$

c++

#include<vector>
using namespace std;
unsigned int findBall(Scales scales) {
    int w = scales.getWeight({0,1,2},{3,4,5});
    if(w == 0) {
        return scales.getWeight({6},{7}) == 1 ? 7 : 6;
    }
    if(w == 1) {
        w = scales.getWeight({3},{4});
        if(w == 0) return 5;
        return w == 1 ? 4 : 3;
    }
    w = scales.getWeight({0},{1});
    if(w == 0) return 2;
    return w == 1 ? 1 : 0;
}