Suuremmat

/**
 * Datat�hti 2021 loppu
 * 358
 * A
 * Suuremmat
 * @author TRS
 */
//Include
#include <bits/stdc++.h>
//Definitions
using namespace std;
//Constants
#define infinity 0x3f3f3f3f
#define linfinity 0x3f3f3f3f3f3f3f3f
#define MOD 1000000007
int t;
unsigned long long n;
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    cin>>t;
    for (int _ = 0; _ < t; _++) {
        cin>>n;
        int maxPowerOf10 = floor(log10(n));
        if (n < 9) {
            cout<<(n + 1)<<"\n";
            continue;
        }
        else if (n < 11) {
            cout<<11<<"\n";
            continue;
        }
        short firstDigit = floor(n / pow(10, maxPowerOf10));
        short secondDigit = (int)floor(n / pow(10, maxPowerOf10 - 1)) % 10;
        if (firstDigit == secondDigit && maxPowerOf10 == 1) {
            if (firstDigit == 9) {
                cout<<"111"<<"\n";
                continue;
            }
            cout<<(firstDigit + 1)<<(firstDigit + 1)<<"\n";
            continue;
        }
        if (firstDigit > secondDigit) {
            for (int d = 0; d <= maxPowerOf10; d++) {
                cout<<firstDigit;
            }
            cout<<"\n";
            continue;
        }
        else if (firstDigit < secondDigit) {
            for (int d = 0; d <= maxPowerOf10; d++) {
                cout<<(firstDigit + 1);
            }
            cout<<"\n";
            continue;
        }
        else { //firstDigit == secondDigit
            vector<int> digits;
            unsigned long long tempn = n;
            for (int d = 0; d <= maxPowerOf10; d++) {
                int digit = tempn % 10;
                digits.push_back(digit);
                tempn /= 10;
            }
            reverse(digits.begin(), digits.end());
            int printFirstDigit = 0;
            int digitsAllSame = 1;
            for (int i = 0; i <= maxPowerOf10; i++) {
                if (digits[i] < firstDigit) {
                    printFirstDigit = 1;
                    digitsAllSame = 0;
                    break;
                }
                else if (digits[i] > firstDigit) {
                    digitsAllSame = 0;
                    break;
                }
            }
            if (digitsAllSame) {
                if (firstDigit == 9) {
                    for (int d = 0; d <= maxPowerOf10 + 1; d++) {
                        cout<<1;
                    }
                }
                else {
                    for (int d = 0; d <= maxPowerOf10; d++) {
                        cout<<(firstDigit + 1);
                    }
                }
            }
            else if (printFirstDigit) {
                for (int d = 0; d <= maxPowerOf10; d++) {
                    cout<<firstDigit;
                }
            }
            else {
                for (int d = 0; d <= maxPowerOf10; d++) {
                    cout<<(firstDigit + 1);
                }
            }
            cout<<"\n";
        }
    }
    return 0;
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED