lavos/cap3/work14.go

package cap3

import "math"

func Work14() []int {
	resultSet := make([]int, 0)
	for num := 100; num <= 999; num++ {
		// 开方取整，再平方判，断是否能获得原来的数，来识别是否为完全平方数
		intSqrt := int(math.Sqrt(float64(num)) + 0.5)
		if intSqrt*intSqrt != num {
			continue
		}

		// num的百位，十位，个位数，其中有相等即可
		a, b, c := num/100, num%100/10, num%10
		if a == b || a == c || b == c {
			resultSet = append(resultSet, num)
		}
	}

	return resultSet
}