每日一道算法题(2)--四键键盘问题

题目

四键键盘:

A键 :在屏幕上显示一个A。

Ctrl_A键:选中整个屏幕。

Ctrl_C键:将选中的区域复制到缓冲区。

Ctrl_V键:将缓冲区的内容输出到光标所在的屏幕位置。

给出步数 n 求出在屏幕中最大的A数量。(n=3时,算法返回3，因为连按三次A键是最优方案;n=7时，算法返回9,最优方案为:A,A,A,Ctrl_A,Crtrl_C,Ctrl_V)

思路分析

    我们首先考虑用动态规划来做,观察题目我们可以发现Ctrl_A,Ctrl_C总是连在一起的,所以我们每次有三种选择：A，Ctrl_A Ctrl_C，Ctrl_V，且需要两个变量分别记录屏幕A的个数和缓冲区A的个数,依次通过选择来改变屏幕A的个数和缓冲区A的个数。

代码展示

#include<stdio.h>

int MaxA(int, int, int);
int Max(int, int, int); 

int main()
{
	int n;
	scanf("%d", &n);
	printf("%d", MaxA(n, 0, 0));
	return 0;
}

int MaxA(int n, int A_num, int copy)
{
	if(n<=0){
		return A_num;
	}else{
		//情况有三种，分别是 A,Ctrl_A Ctrl_c, Ctrl_V; 
		return Max(
			MaxA(n - 1, A_num + 1, copy),
			MaxA(n - 2, A_num, A_num),
			MaxA(n - 1, A_num + copy, copy)
		);
	}
}

int Max(int a, int b, int c)//找出最大的数 
{
	if(a >= b){
		if(a >= c){
			return a;
		}else{
			return c;
		}
	}else{
		if(b >= c){
			return b;
		}else{
			return c;
		}
	}
}

    通过递归我们应该都能理清了思路,但是此代码的时间复杂度超过了O(n^2),算法不是很完美。

    每次最优方案的最后要么是A，要么是Ctrl_V,所以我们能否减少选择的种类来减低的时间复杂度。

优化代码如下:

#include<stdio.h>

int MaxA(int);

int main()
{
	int n;
	scanf("%d", &n);
	printf("%d", MaxA(n));
	return 0;
}

int MaxA(int n)
{
	int data[n + 1];//data[0]位置不放数据
	data[0] = 0;//初始化(从0开始)
	int i, j; 
	for(i = 1; i <= n; i++){
		//情况一 :按A键(增加一个A)
		data[i] = data[i - 1] + 1;
		//情况二: 按Ctrl_V键(增加的A的个数不确定)
		for(j = 2; j < i; j++){
		//复制的个数在改变
		/*(i - j + 1)中的+1是为了算上j开始时的位置的A的个数,
		且起始位置A的个数就为data[j - 2],因为data[j-1],data[j]两代表的是Ctrl_A,Ctrl_C两步,
		data[j]后的项代表的是Ctrl_V,最后就可以算出个数。*/ 
		data[i] = (data[i] >= data[j - 2] * (i - j + 1)) ? data[i] : data[j - 2] * (i - j + 1);//取两者较大的值
		}
	}
	return data[n];
}

    首先创建了一个数组用来记录每步中A的个数,通过循环来计算A的个数。在循环中分两种情况。情况二会复杂一些。

    此优化代码的空间复杂度为O(n),时间复杂度为:O(n^2)。