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 (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 - 1] != s[i]) ok = 0;
 
    if (ok)
    {
        for (int i = 0; i < n; ++i)
            cout << char(s[i] + 1);
        cout << '\n';
        return;
    }
 
    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: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED