Greater Integers

#include <bits/stdc++.h>
 
using namespace std;
 
void solve(){
	int64_t n;
	cin >> n; 
	vector<int> digit; 
	int64_t p = n; 
	while(n){
		digit.push_back(n%10); 
		n /= 10; 
	}
 
	reverse(digit.begin(), digit.end()); 	
	int x = digit.size(); 
 
	int64_t ans = 0, cur = 1; 
	for(int i = 0; i < x; i++){
		ans += cur;
		cur *= 10; 
	}
	// cout << ans*digit[0] << '\n';
	if(ans*digit[0] <= p){
		if(digit[0] == 9){
			ans += cur; 
			cout << ans << '\n';
		}
		else{
			cout << ans*(digit[0] + 1) << '\n'; 
		}
	}
	else cout << ans*digit[0] << '\n'; 
}
int32_t main(){
	ios_base::sync_with_stdio(false); 
	cin.tie(NULL); 
	
	int t;
	cin >> t; 
 
	for(int i = 0; i < t; i++){
		solve(); 
	}	
	return 0; 
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED