Greater Integers

#include<bits/stdc++.h>
using namespace std;
int lst(int n)
{
while(n)
{
if(n<10)
return n%10;
n/=10;
}
return n;
}
int calc(int di,int n)
{
int nc(0),sm(0);
while(n)
{
++nc;
n/=10;
}
for(int i=0; i<nc; ++i)
sm+=pow(10,i)*di;
return sm;
}
int main()
{
int t;
cin>>t;
while(t--)
{
long long int n;
cin>>n;
if(n<9)
cout<<n+1<<"\n";
else
{
int nm=calc(lst(n),n);
if(n<nm)
cout<<nm<<"\n";
else if(n==9)
cout<<11<<"\n";
else
{
int an=lst(n)+1;
if(an==10)
an=1;
while(n)
{
cout<<an;
n/=10;
}
if(an==1)
cout<<1;
cout<<"\n";
}
}
}
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: WRONG ANSWER