HJ99 题解 | #自守数#

#include <iostream>

#include <string>

#include <algorithm>

using namespace std;

bool isZiShouShu(long n)

{

    if (n == 0 || n == 1) {

        return true;

    }

    long m = n * n;

    string str1 = std::to_string(n);

    string str2 = std::to_string(m);

    string str3 = str2.substr(str2.size() - str1.size());

    return str3 == str1;

}

int GetZiShouCount(long n)

{

    int count = 0;

    for (long i = 0; i <= n; i++) {

       if (isZiShouShu(i)) {

           count++;

       }

    }

    return count;

}

int main() {

    long n;

    while (cin >> n) { // 注意 while 处理多个 case

        cout << GetZiShouCount(n) << endl;

    }

}