ugly_number_II

hitzzc · hitzzc · commit 7c449628d4b7 · 2016-10-21T17:52:41.000+08:00
diff --git a/README.md b/README.md
@@ -202,6 +202,7 @@ Golang solution for leetcode. For each problem, there is a simple *_test.go to t
 #### [258. Add Digits](https://github.com/hitzzc/go-leetcode/tree/master/add_digits)
 #### [260. Single Number III](https://github.com/hitzzc/go-leetcode/tree/master/single_number_III)
 #### [263. Ugly Number](https://github.com/hitzzc/go-leetcode/tree/master/ugly_number)
+#### [264. Ugly Number II](https://github.com/hitzzc/go-leetcode/tree/master/ugly_number_II)
 
 
 
diff --git a/ugly_number_II/ugly_number_II.go b/ugly_number_II/ugly_number_II.go
@@ -0,0 +1,30 @@
+package ugly_number_II
+
+func nthUglyNumber(n int) int {
+	if n == 1 {
+		return 1
+	}
+	records := []int{1}
+	i, j, k := 0, 0, 0
+	for len(records) < n {
+		next := min(2*records[i], min(3*records[j], 5*records[k]))
+		if next == 2*records[i] {
+			i++
+		}
+		if next == 3*records[j] {
+			j++
+		}
+		if next == 5*records[k] {
+			k++
+		}
+		records = append(records, next)
+	}
+	return records[len(records)-1]
+}
+
+func min(i, j int) int {
+	if i < j {
+		return i
+	}
+	return j
+}
