Greater Integers

#include <bits/stdc++.h>
using namespace std;
using int128 = __int128;
using int64 = long long;
int main() {
  ios::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);
  set<int128> s;
  for (int i = 1; i <= 9; ++i) {
    for (int len = 1; len <= 20; ++len) {
      int128 num = 0;
      for (int j = 0; j < len; ++j) {
        num *= 10LL;
        num += i;
      }
      s.insert(num);
    }
  }
  int tc; cin >> tc;
  while (tc--) {
    int64 n; cin >> n;
    int128 ans = *s.upper_bound((int128)n);
    vector<int> res;
    while (ans) {
      res.push_back(ans%10);
      ans /= 10;
    }
    reverse(res.begin(), res.end());
    for (auto x : res) cout << x;
    cout << '\n';
  }
  return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED