HJ99 自守数(c++/python)

c++:

#include
using namespace std;

int main(){
    int n;
    cin >> n;
    int count = 0;
    for(int i = 0; i <= n; i++){
        int j = i;
        int pow_i = i*i;
        while(j){
            if(j%10 == pow_i%10){
                j = j/10;
                pow_i = pow_i/10;
                
            }else{
                break;
            }
        }
        if(j == 0){
            count++;
        }
    }
    cout << count;
}

python:

while True:
    try:
        n = int(input())
        cnt = 0
        for i in range(n+1):
            if(str(i*i).endswith(str(i))):
                cnt += 1
        print(cnt)
    except:
        break