Greater Integers

#include <bits/stdc++.h>

using namespace std ;

const long long MAX = 2e18 ;

long long n ;

vector<long long>v ;

int main()
{
	ios_base::sync_with_stdio(0) ;
	cin.tie(0) ;
	for(int i = 1 ; i <= 9 ; ++i)
	{
		for(long long j = i ; j < MAX ; j = j * 10ll + i)
			v.push_back(j) ;
	}
	sort(v.begin() , v.end()) ;
	int t ;
	cin>>t ;
	while(t--)
	{
		cin>>n ;
		int idx = upper_bound(v.begin() , v.end() , n) - v.begin() ;
		cout<<v[idx]<<"\n" ;
	}
	return 0 ;
}		

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: WRONG ANSWER