func judgeSquareSum(c int) bool { for a := 0; a*a <= c; a++ { b2 := c - a*a if b := int(math.Sqrt(float64(b2))); b*b == b2 { return true } } return false }