Greater Integers

#include<bits/stdc++.h>
#define ll long long
#define inf 1e17
using namespace std;
 
void solve(){
    string s;
    cin >> s;
    bool flag = true;
    ll n = s.size();
    for(int i = 0; i < n; i++){
        if(s[i] != '9'){
            flag = false;
            break;
        }
    }
    if(flag){
        string ans = "";
        for(int i = 0; i <= n; i++){
            ans += '1';
        }
        cout << ans << "\n";
        return;
    }
    if(s[0] == '9'){
        for(int i = 0; i < n; i++){
            cout << 9;
        }
        cout << "\n";
        return;
    }
    string ans = "";
    for(int i = 0; i < n; i++){
        ans += s[0];
    }
    if(ans > s){
        cout << ans << "\n";
        return;
    }
    ans.clear();
    s[0]++;
    for(int i = 0; i < n; i++){
        ans += s[0];
    }
    cout << ans << "\n";
}
int main(){
	ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    int t = 1;
    cin >> t;
    while(t--){
        solve();
    }
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED