Greater Integers

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

using namespace std;
using namespace __gnu_pbds;
 
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define fbo find_by_order
#define ook order_of_key
 
typedef long long ll;
typedef pair<int,int> ii;
typedef vector<int> vi;
typedef long double ld; 
typedef pair<ld,ld> state;
typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> pbds;

vector<ll> vec;

void solve()
{
	ll x; cin>>x;
	int lb=upper_bound(vec.begin(),vec.end(),x)-vec.begin();
	cout<<vec[lb]<<'\n';
}

ll num(int d, int t)
{
	ll ans=0;
	for(int i=0;i<t;i++)
	{
		ans*=10; ans+=d;
	}
	return ans;
}

int main()
{
	ios_base::sync_with_stdio(0); cin.tie(0);
	int t; cin>>t;
	for(int t=1;t<=18;t++)
	{
		for(int d=1;d<=9;d++) vec.pb(num(d,t));
	}
	vec.pb(num(1,19));
	while(t--) solve();
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED