Suggestion for solution 202

leetcode/0202.Happy-Number/202. Happy Number.go

@@ -1,27 +1,32 @@
 package leetcode
 
-func isHappy(n int) bool {
-	if n == 0 {
-		return false
-	}
-	res := 0
-	num := n
-	record := map[int]int{}
-	for {
-		for num != 0 {
-			res += (num % 10) * (num % 10)
-			num = num / 10
-		}
-		if _, ok := record[res]; !ok {
-			if res == 1 {
-				return true
-			}
-			record[res] = res
-			num = res
-			res = 0
-			continue
-		} else {
-			return false
-		}
+func getSquareOfDigits(n int) int {
+	squareOfDigits := 0
+	temporary := n
+
+	for temporary != 0 {
+		remainder := temporary % 10
+		squareOfDigits += remainder * remainder
+		temporary /= 10
 	}
+
+	return squareOfDigits
+}
+
+func isHappy(n int) bool {
+	record := map[int]int{}
+
+	for n != 1 {
+		record[n] = n
+
+		n = getSquareOfDigits(n)
+
+		for _, previous := range record {
+			if n == previous {
+				return false
+			}
+		}
+	}
+
+	return true
 }