Greater Integers

#include <bits/stdc++.h>
 
using namespace std;
 
#define f first
#define s second
 
typedef long long ll;
typedef pair<int,int> pi;
 
const ll large = 2e18;
 
int main(){
    ios_base::sync_with_stdio(0); cin.tie(0);
    int t; cin >> t;
    vector<ll> nums;
    for(int i = 1; i < 10; i++){
        ll cur = i;
        nums.push_back(cur);
        for(int j = 1; j < 18+(i == 1); j++){
            cur *= 10LL;
            cur += (ll)i;
            nums.push_back(cur);
        }
    }
    sort(nums.begin(), nums.end());
    while(t--){
        ll n; cin >> n;
        for(auto &a : nums){
            if(a > n){
                cout << a << "\n";
                break;
            }
        }
    }
    return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED