题解 | #自守数#

package main

import (
	"fmt"
)

func isSelfDefineNumber(n int) bool {
    nn := n * n 
    for n > 0 {
        if n % 10 != nn % 10 {
            return false
        }

        n /= 10
        nn /= 10
    }

    return true
}

func main() {
    var n int
    fmt.Scan(&n)

    var cnt int
    for i:=0; i<=n; i++ {
        if isSelfDefineNumber(i) {
            cnt++
        }
    }

    fmt.Println(cnt)
}

// 本题输入一个整数，所以采用：fmt.Scan(&n)