Codeforces · 2185C

Shifted MEX

This C++ solution uses sorting & greedy for Codeforces 2185C Shifted MEX. Read the reasoning, inspect the code, or try your own test case below.

2185CImplementation & simulationSorting & greedyC++40 lines
Solution183of 248
Open official problem ↗ Download C++ file ↓ Search the library → Open full Code Lab ↗ Report an issue ↗

Approach

Sorting & greedy

Shifted MEX matches the shift choices and maximum-MEX computation in the implementation.

Implementation & simulation

Problem and code

Useful links.

Written by benbenyaojifen. Try the problem first, then compare your approach with the code.

Open official problem ↗View exact source file ↗
Implementation

C_Shifted_MEX.cpp

C++

    #include <bits/stdc++.h>
    using namespace std;
    typedef long long ll;
    const int inf = 1e9;
    const long long INF = 1e17; //❄️
    void solve() {
        int n; cin >> n;
        vector<int> v(n);
        for (int i = 0; i < n; i++) {
            cin >> v[i];
        }
        int ans = 0;
        auto mex = [&](vector<int>& a) {
            unordered_set<int> s;
            for (int i = 0; i < a.size(); i++) {
                s.insert(a[i]);
            }
            for (int i = 0; ; i++) {
                if (!s.count(i)) return i;
            }
        };
        sort(v.begin(), v.end());
        for (int i = 0; i < n; i++) {
            int diff = 0 - v[i];
            vector<int> tmp(n);
            for (int j = 0; j < n; j++) {
                tmp[j] = v[j] + diff;
            }
            ans = max(ans, mex(tmp));
        }
        cout << ans << '\n';
    }
    int main() {
        ios::sync_with_stdio(0); cin.tie(0); 
        int t; cin >> t;
        while (t--) {
            solve();
        }
        return 0;
    }
        

Did this explanation save you time? I'm a Grade 11 student building this free library to make difficult algorithms easier to understand.

Buy me a coffee ↗Keep studying →

Test this problem

Run your code here.

Paste your code, run a test case, compare the output, or trace selected values.

Full trace, comparison & stress testing ↗
StatusReady
Output
No run yet.
Diagnostics
No diagnostics yet.

Each run is isolated and has strict limits. Passing one test does not guarantee the judge will accept the solution.