Greater Integers

#include <bits/stdc++.h>
using namespace std;
#define int long long
 
signed main(){
    ios_base::sync_with_stdio(false); cin.tie(nullptr);
    int tt; cin >> tt;
    while(tt--) {
        int n;
        cin >> n;
        int m = n;
        int d = 0, last = 0;
        while (m) {
            ++d;
            last = m;
            m /= 10;
        }
        if (last == 9) {
            string s, t = "1";
            int comp = 0;
            for (int i = 0; i < d; ++i) {
                s += '9';
                t += '1';
                comp *= 10;
                comp += 9;
            }
            if (comp<=n) {
                cout << t << "\n";
            } else cout << s << "\n";
        } else {
            int comp = 0;
            for (int i = 0; i < d; ++i) {
                comp *= 10;
                comp += last;
            }
            if (comp<=n) {
                for (int i = 0; i < d; ++i) {
                    cout << last + 1;
                }
                cout << "\n";
            } else {
                for (int i = 0; i < d; ++i) {
                    cout << last;
                }
                cout << "\n";
            }
        }
    }
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED