题目:正整数序列Q中的每个元素都至少能被正整数a和b中的一个整除,现给定a和b,需要计算出Q中的前几项,例如,当a=3,b=5,N=6时,序列为3,5,6,9,10,12
(1)设计一个函数void generate(int a, int b, int N, int * Q)计算Q的前N项
(2)设计测试数据来验证函数程序在各种输入下的正确性
最简单的想法就是,从1开始挨个验证,能否被a或b整除,直到找到N项为止。但是这个方法如果a和b很大,那么就要遍历很多数字,而且模运算也比较耗时。
实现代码如下:
void generate(int a, int b, int N, int * Q)
{
if(a <= 0 || b <= 0 || N < 0 || Q == NULL)
throw new exception("Invalid input!");
int cnt = 0;
int i = 1;
while(cnt < N)
{
if(i % a == 0 || i % b == 0)
Q[cnt++] = i;
i++;
}
return;
}
优化思路,我们只需要从小到大输出a的倍数或b的倍数就行,使用两个变量x和y,x从a开始以a为步长增长,y从b开始以b为步长增长,每次输出较小者且较小者按照相应步长增长,当x==y时,同时增长。
实现代码如下:
void generate(int a, int b, int N, int * Q)
{
if(a <= 0 || b <= 0 || N < 0 || Q == NULL)
throw new exception("Invalid input!");
int cnt = 0;
int x = a;
int y = b;
while(cnt < N)
{
if(x == y)
{
Q[cnt++] = x;
x += a;
y += b;
}
else if(x < y)
{
Q[cnt++] = x;
x += a;
}
else
{
Q[cnt++] = y;
y += b;
}
}
return;
}
测试数据如:
a = 1, b = 2
a = 2, b = 1
a = 0, b = -1
a = 3, b = 5
a = 5, b = 3
a = 7, b = 7
a = 9997, b = 13337
a = 13337, b = 9997