Greater Integers

#include <bits/stdc++.h>
using namespace std;
 
#define int long long
#define all(x) (x).begin(),(x).end()
#define pb push_back

signed main(){
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  int t; cin >> t;
  while(t-- ){
    int n; cin >> n;
    bool a9 = 1;
    int cn = n;
    int amt = 0;
    int lst = 0;
    while(cn) {
      ++amt;
      a9 &= (cn % 10) == 9;
      lst = cn % 10;
      cn /= 10;
    }
    if(a9) {
      for(int i = 0; i <= amt; ++i) {
        cout << 1;
      }
      cout << "\n";
      continue;
    }
    bool al = 1, as = 1;
    cn = n;
    while(cn) {
      al &= (cn % 10) <= lst;
      as &= (cn % 10) == lst;
      cn /= 10;
    }
    if(al && !as) {
      for(int i = 0; i < amt; ++i) {
        cout << lst;
      }
      cout << "\n";
    } else {
      for(int i = 0; i < amt; ++i) {
        cout << lst + 1;
      }
      cout << "\n";
    }
  }
}

Test 1

Group: 1, 2

Verdict: WRONG ANSWER

Test 2

Group: 2

Verdict: WRONG ANSWER