题目描述
小Q在进行一个对数字进行拆分的游戏,游戏规则如下
小Q最初只有一个整数N,接下来每一轮中,小Q被允许对现有的每个数进行下面两个操作之一
- 1、对当前小Q手里的所有数减1
- 2、把所有数都拆分为更小的两个数之和
但是拆分操作只允许使用至多k次,现在小Q想知道把N完全消去需要多少轮操作。
输入描述
输入一行包含两个整数N, K (1< N< 100,0< K<100)
输出描述:
输出一个整数,表示至少需要的轮数。
一、思路
对小Q手里的每一个数进行二拆分,一直到K=0或者出现1为止:
-
拆分出现了1,因为1不可拆分,所以执行减1操作
注:当拆分到出现1的时候,此时最大的数也不过是2,为什么?
因为实际你可以将这个过程视为一颗二叉树不断均匀分裂的过程,此时:左右子树之差不超过1。
假设N拆分(分裂)成了N1与N2,那么N1与N2继续往下拆分(分裂),因为N1与N2之差不超过1,所以各自分裂形成二叉树的高度最多也就相差1。 -
K=0,即使未完全拆分,只要拆分次数一致,那么底层的最大最小数之差,一定是1或者0
综上所述,可以通过设置一个标志位,来查看是否有奇数出现(出现奇数,分裂后的树高相差1)
C++代码:
#include <iostream>
#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <string>
#include <stack>
using namespace std;
class Solution {
public:
int splitIntger(int N, int K) {
int num = N, i = 0, flag = 0;
while (num > 1 && K != 0) {
i++;
K--;
if (num % 2 == 1)
flag = 1;
num = num / 2;
}
i = i + num + flag;
return i;
}
};
int main()
{
int N, K;
Solution a;
cin >> N;
cin >> K;
cout << a.splitIntger(N, K) << endl;
}