srm549.cpp

class BallAndHats
{
public:
  int getHat(string hats, int numSwaps)
    {
      int idx = hats.find("o");
      if (numSwaps == 0) return idx;
      return (idx + (numSwaps % 2)) % 2;
    }
};