Solution Class Reference

Public Member Functions
vector< int >	queryResults (int limit, vector< vector< int > > &queries)

Detailed Description

Definition at line 5 of file main.cpp.

Member Function Documentation

queryResults()

vector< int > Solution::queryResults ( int  limit, vector< vector< int > > &  queries  )

inline

Definition at line 8 of file main.cpp.

    {
        limit = limit; // not needed
        vector<int> result;
        unordered_map<int, int> balls;
        unordered_map<int, int> colors;
        unordered_set<int> distinct_colors;
        
        for (const auto& query : queries)
        {
            int b = query[0], c = query[1];
 
            // ball already painted
            if (balls.count(b))
            {
                int old_color = balls[b];
                if (--colors[old_color] == 0)
                    distinct_colors.erase(old_color);
            }
 
            // assigning new color
            balls[b] = c;
            if (++colors[c] == 1)
                distinct_colors.insert(c);
 
            result.push_back(distinct_colors.size());
        }
        
        return result;
    }

Referenced by main().

---

The documentation for this class was generated from the following file:

• main.cpp