Leetcode-650 只有两个键的键盘

描述

最初在一个记事本上只有一个字符 'A'。你每次可以对这个记事本进行两种操作：

Copy All (复制全部) : 你可以复制这个记事本中的所有字符(部分的复制是不允许的)。
Paste (粘贴) : 你可以粘贴你上一次复制的字符。
给定一个数字 n 。你需要使用最少的操作次数，在记事本中打印出恰好 n 个 'A'。输出能够打印出 n 个 'A' 的最少操作次数。

示例 1

输入: 3
输出: 3
解释:
最初, 我们只有一个字符 'A'。
第 1 步, 我们使用 Copy All 操作。
第 2 步, 我们使用 Paste 操作来获得 'AA'。
第 3 步, 我们使用 Paste 操作来获得 'AAA'。

说明

n 的取值范围是 [1, 1000] 。

来源：力扣（LeetCode）
链接：https:leetcode-cn.com/problems/2-keys-keyboard
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

解法

//步骤性问题
//分类关键是下一步操作
//状态变量：a(当前a的数量), c(当前剪切板a的数量)

const (
	COPY_ALL = 0
	PASTE = 1
	INF = math.MaxInt32
)

func minSteps(n int) int {
	if n == 1 {
		return 0
	}
	return Op(n, 1, 0, 0, COPY_ALL)
}

// a: A目前的数量
// c: 剪切板A的数量
// t: 操作次数
// r: A的目标数量
// code: 下一步操作
func Op(r, a, c, t, code int) int {
	t+=1
	if code == COPY_ALL {
		// copy all
		c = a
	}
	if code == PASTE {
		// paste
		a += c
	}
	if a >= r {
		if a > r {
			return INF
		} else {
			return t
		}
	}
	if code == COPY_ALL {
		return Op(r, a, c, t, PASTE)
	} else {
		return min(Op(r, a, c, t, COPY_ALL), Op(r, a, c, t, PASTE))
	}
}

 

优化

// 化为数学问题
// (CPP)(CPPPP)(CPPPPP)
// 3个A, 3*5 个A, 3*5*6个A
// 3+5+6 = 14 次
// 6 = 2*3 > 2+3
// 3+5+2+3 = 13 次
func minStepsOthers(n int) int {
	res := 0
	d := 2
	// n= 2^(k1)*3^(k2)*5^(k3)....
	for n > 1 {
		for n % d == 0 {
			// n = ...* d^(k)
			res += d
			n /= d
		}
		// 如果这里是合数，不会进入上述循环
		d += 1
	}
	return res
}