Greater Integers

#include <bits/stdc++.h>

#define pb push_back
#define int long long
#define all(x) x.begin(), x.end()
#define ld long double
using namespace std;

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    vector<int> nums;
    for (int len = 1; len <= 18; len++) {
        for (int d = 1; d <= 9; d++) {
            int s = 0;
            for (int _ = 0; _ < len; _++) {
                s *= 10;
                s += d;
            }
            nums.pb(s);
        }
    }
    int t;
    cin >> t;
    while (t--) {
        int n;
        cin >> n;
        cout << *upper_bound(all(nums), n) << '\n';
    }

}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: WRONG ANSWER