[HDOJ]1005. Number Sequence
数字游戏,如果硬说它有算法的话,那也只能说去找规律了。
初看本题的人,会觉得这这是一个简单的递归题目,其实我一开始也是这么想的,但后来提交了如下代码后我才明白。
using namespace std;
int main()
{
int a,b,n;
while(cin>>a>>b>>n && n)
{
int f1 = 1,f2 = 1,f,t = 3;
while(t <= n)
{
f = (a*f2 + b*f1)%7;
f1 = f2;
f2 = f;
++t;
}
cout<<f<<endl;
}
return 0;
}
细细看下题目中的条件要求:1 <= n <= 100,000,000,你就应该明白本题绝不是简单的递归就可以解决的,上述代码的答案是TLE.(后来加):其实看到题目中出现求余运算,我就应该很快的反应过来,这道题目绝对不是简单的递归运算就可以解决问题的,因为求余运算会有一定的重复,所以,本题中的数据一定会出现重复,从重复中找到规律,问题就可以解决了。
其实解决本题的方法是找出规律,当f1,f2和a b的值确定后,你就可以利用题目中给出的公式往后计算了,如果我们把所有fn的取值算作一个队列(非数据结构中的队列)的话,可以肯定的是,这个队列不会延展太长,到了一定长度,肯定会出现重复,而我们需要做的就是找出这个重复点。
什么时候会出现重复呢,如果我们假定队列是如下循环的:
f1 = f2;
f2 = f;
那么很简单就可以推出:当f1 = 1,且f2 = 1时就会出现队列一开始的循环了。此时我们就不需要继续向下递归了。不过我们需要把出现循环以前的所有值都保存下来。相比于时间复杂度来说,这点空间复杂度还是值得的。
这样写了代码,结果却Memory Limit Exceeded,郁闷,今天太晚了,明天再搞。
#include <vector>
using namespace std;
int main()
{
int a,b,n;
while(cin>>a>>b>>n && n)
{
vector<int> ivec;
int f1 = 1,f2 = 1,f,t = 3,count,num;
ivec.push_back(f1);
ivec.push_back(f2);
while(1)
{
f = (a*f2 + b*f1)%7;
f1 = f2;
f2 = f;
if(f1 == 1 && f2 == 1)
break;
else
ivec.push_back(f);
++t;
}
count = t - 2;
num = n%count;
if(num == 0)
num = count;
cout<<ivec.at(num - 1)<<endl;
ivec.clear();
}
return 0;
}
昨天晚上找了杭电的WSN大牛问了下,一开始他也不是很清楚,只是让我把vector<int> ivec;设置为全局变量试下,这个我早已经试过,结果还是Memory limit Exceeded,问题应该不是出在这里,我初步猜测应该是有一组测试数据,让vector<int> ivec一直在执行push_back()操作,所以才会出现上述错误,看了WSN的AC代码后,他与我想法的区别在于数字的重复不一定是我所想的从头开始的重复,也可能是在中间部分开始重复的,举个简单的例子,这里没有理由依据:1 1 2 5 6 5 4 4 8 6 2 5 4....,如果是这样下去的话,从5 和 4开始就是重复了,并不是从开始的1 和1 开始重复的,这样也有道理。而且也实际上应该这么考虑才对。
后来,我随便试了几组数据,就发现出现了问题,当我输入21 56 45这组数据时,上面的程度没有输出,Debug后才知道,其实不是没有输出,而是程序在while(1)处进入了死循环,这样就导致我的ivec一直不间断的进行push_back()操作,才会有了开始的Memory Limit Exceeded这种错误。既然找到了错误,下面就剩下修改代码了,我决定采用WSN的想法,在判断是否开始重复时多加入一个条件判断,修改后AC,代码及注释如下:
#include <vector>
using namespace std;
vector<int> ivec;
//布尔数组元素flag[i][j]如果为true,则说明前面已经出现了i 和 j两者的组合
bool flag[7][7];
void init(void)
{
int i,j;
for(i = 0;i < 7;++i)
for(j = 0;j< 7;++j)
flag[i][j] = false;
}
int main()
{
int a,b,n;
while(cin>>a>>b>>n)
{
if(a == 0&&b == 0&&n == 0)
break;
ivec.clear();
init();
ivec.push_back(1);
ivec.push_back(1);
flag[1][1] = true;
int count = 1,f;
while(1)
{
f = (a*ivec.at(count)%7 + b*ivec.at(count - 1)%7)%7;
ivec.push_back(f);
++count;
//如果flag变量为true,则说明前面已经出现了这两者的组合,出现重复,无需下一步计算,直接break退出即可
if(flag[ivec.at(count)][ivec.at(count - 1)] == true)
break;
else
flag[ivec.at(count)][ivec.at(count - 1)] = true;
}
//count中存放的是ivec中出现循环前的元素总个数,注意ivec中的下标是从0开始计数的
count = count - 1;
if(n < count)
cout<<ivec.at(n-1)<<endl;
else
{
int j;
//for循环的目的是找出从那个地方开始重复,此处应该是从j处开始循环,注意j是从0下标开始计数的
for(j = 0;;++j)
if(ivec.at(count) == ivec.at(j) && ivec.at(count + 1) == ivec.at(j+1))
break;
n = (n - j)%(count - j);
if(n == 0)
n = count - j;
n += j;
cout<<ivec.at(n-1)<<endl;
}
}
return 0;
}
后记:在我解决我这道题出现问题的过程中,我也在网上搜索了很多,找到了一些代码去测试,结果发现其实杭电服务器上关于本题的测试数据是不完整的,就拿我刚才那个测试数据来说:21 56 n来说,其循环应该为:
1 1 0 0 0 0 0 0 0 ...........,而我在网上找到的一些代码,比如下面这个代码:(来源:http://hi.baidu.com/chenghui2050/blog/item/84c552ad3124ee0f4a36d660.html)
using namespace std;
int main()
{
int a,b,i;
long long f[55],n;
while(cin>>a>>b>>n)
{
if(a==0&&b==0&&n==0)break;
f[1]=f[2]=1;
for(i=3;i<=49;i++)
f[i]=(a*f[i-1]+b*f[i-2])%7;
cout<<f[n%48]<<endl;
}
return 0;
}
按照WSN的思路考虑的话,应该是考虑比较全面的了,赞the lord of WSNs,呵呵.
这个问题先告一段落了,残酷的测试数据,倒。。。