Greater Integers

#include <bits/stdc++.h>

using namespace std;

#define int long long
#define ff first
#define ss second
#define pb push_back

const int N = 2e5 + 5, inf = 1e18 + 100;

void solve()
{
    string s;
    cin >> s;

    int n = s.length();
    if (n == 1)
    {
        if (s == "9") cout << 11;
        else cout << char(s[0] + 1);
        cout << '\n';
        return;
    }

    if (s[0] == '9')
    {
        int ok = 1;
        for (int i = 1; i < n; ++i)
            if (s[i - 1] != s[i]) ok = 0;

        if (ok)
        {
            for (int i = 1; i <= n + 1; ++i)
                cout << 1;
        }
        else
        {
            for (int i = 1; i <= n; ++i)
                cout << 9;
        }
        cout << '\n';
        return;
    }

    int ok = 1;
    for (int i = 1; i < n; ++i)
    {
        if (s[i] == s[0]) continue;
        ok = s[i] < s[0];
        break;
    }

    if (ok)
    {
        for (int i = 0; i < n; ++i)
            cout << s[0];
    }
    else
    {
        for (int i = 0; i < n; ++i)
            cout << char(s[0] + 1);
    }
    cout << '\n';
}

signed main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int q;
    cin >> q;
    while (q--)
        solve();
    return 0;
}

Test 1

Group: 1, 2

Verdict: WRONG ANSWER

Test 2

Group: 2

Verdict: ACCEPTED