LeetCode 1538. Guess the Majority in a Hidden Array

Example 1:
Input: nums = [0,0,1,0,1,1,1,1]
Output: 5
Explanation: The following calls to the API
reader.length() 
// returns 8 because there are 8 elements in the hidden array.
reader.query(0,1,2,3) 
// returns 2 this is a query that compares the elements nums[0], nums[1], nums[2], nums[3]
// Three elements have a value equal to 0 and one element has value equal to 1 or viceversa.
reader.query(4,5,6,7) 
// returns 4 because nums[4], nums[5], nums[6], nums[7] have the same value.
we can infer that the most frequent value is found in the last 4 elements.
Index 2, 4, 6, 7 is also a correct answer.

Example 2:
Input: nums = [0,0,1,1,0]
Output: 0

Example 3:
Input: nums = [1,0,1,0,1,0,1,0]
Output: -1
 
Constraints:
5 <= nums.length <= 10^5
0 <= nums[i] <= 1

/**
 * // This is the ArrayReader's API interface.
 * // You should not implement it, or speculate about its implementation
 * class ArrayReader {
 *   public:
 *     // Compares 4 different elements in the array
 *     // return 4 if the values of the 4 elements are the same (0 or 1).
 *     // return 2 if three elements have a value equal to 0 and one element has value equal to 1 or vice versa.
 *     // return 0 : if two element have a value equal to 0 and two elements have a value equal to 1.
 *     int query(int a, int b, int c, int d);
 *
 *     // Returns the length of the array
 *     int length();
 * };
 */

class Solution {
public:
    int guessMajority(ArrayReader &reader) {
        int n = reader.length();
        int start = reader.query(0,1,2,3);
        int g1 = 1, g2 = 0, idx1 = 0, idx2 = -1;
		//假设idx = 0 的数是第一类，其个数为 g1 = 1
        for(int i = 4; i < n; ++i)
        {
            if(reader.query(1,2,3,i)==start)//0， i 是否是一类
                g1++;
            else
                g2++, idx2 = i;
        }//0 与 4,5...n-1 是否是一类
        //还要确定1,2,3
        int q = reader.query(0,2,3,4);
        int p = reader.query(1,2,3,4);
        if(q == p)//0和1是一类
            g1++;
        else
            g2++, idx2 = 1;
        q = reader.query(0,1,3,4);
        // p = reader.query(1,2,3,4);
        if(q == p)//0和2是否是一类
            g1++;
        else
            g2++,idx2 = 2;
        q = reader.query(0,1,2,4);
        // p = reader.query(1,2,3,4);
        if(q == p)//0和3是否是一类
            g1++;
        else
            g2++,idx2 = 3;
        if(g1 == g2) return -1;
        if(g1 > g2) return idx1;
        return idx2;
    }
};