NYOJ 639题

此题原本就是个水水题，不过却也耽误了半个多小时才A过去，在此先把自己的代码贴出来，欢迎大家提出更好的想法……

题目：

找规律

时间限制： 1000 ms  |           内存限制： 65535 KB

难度： 2

描述

xiaod 在小学时最喜欢上数学课了，有一次上课，数学老师布置了一道题目：给你一组有规律的整数，而且只给出前5项，让你求出后面5项。比如：1,2,3,4,5,__,__,__,__,__。显然这是等差数列，所以答案应该是6,7,8,9,10。为了简化题目，老师规定规律只有三种：等差数列、等比数列、斐波那契数列（f[i]=f[i-2]+f[i-1]）；且公差、公比均为整数，所有出现的数字不会超出 int 范围。怎么样，很简单吧？聪明的你，来比一比你和 xiaod 谁算的快吧。

输入

有多组测试数据。
每行5个整数，代表题目给出的前5项。

输出

如果能找到符合条件的规律，则输出5个整数，代表后5项的值。
否则，请输出"None"。（引号不用输出）

样例输入

1 2 3 4 5
2 3 5 8 13

样例输出

6 7 8 9 10
21 34 55 89 144

我的代码：

#include<iostream>
#include<cstdio>
using namespace std;
//此题真是变态，首先要注意判断等比数列时要注意必须5个数全不为零，才能避免求公比时候有除零错误
//其次在判断等差数列时要注意先把任意两个数字的差求出来再把每两个数进行比较
//而是用 if(a[1]-a[0]==a[2]-a[1] && a[2]-a[1]==a[3]-a[2] && a[3]-a[2]== a[4]-a[3])也可以
//但是是用if(a[1]-a[0]==a[2]-a[1]==a[3]-a[2]== a[4]-a[3])就不行了
int main()
{
    int a[20];
    int d,q,i;

    while(scanf("%d %d %d %d %d",&a[0],&a[1],&a[2],&a[3],&a[4]) != EOF)
    {
        int num = 0;
        for(i = 0;i < 5;i++)
        {
            if(a[i] != 0)
                num++;
        }

        if(a[1]-a[0]==a[2]-a[1] && a[2]-a[1]==a[3]-a[2] && a[3]-a[2]== a[4]-a[3])
        {
            d = a[2]-a[1];
            printf("%d %d %d %d %d\n",a[4] + d,a[4] + 2*d,a[4] + 3*d,a[4] + 4*d,a[4] + 5*d);
        }
        else if(a[0]+a[1]==a[2] && a[1]+a[2]==a[3]&& a[2]+a[3]==a[4] )
        {
             a[5] = a[3] + a[4];
             a[6] = a[4] + a[5];
             a[7] = a[5] + a[6];
             a[8] = a[6] + a[7];
             a[9] = a[7] + a[8];
             printf("%d %d %d %d %d\n",a[5],a[6],a[7],a[8],a[9]);
        }
        else if(num == 5 && a[1]*a[1] == a[0]*a[2] && a[2]*a[2] == a[1]*a[3] && a[3]*a[3] == a[2]*a[4])
        {
             q = a[1]/a[0];
             printf("%d %d %d %d %d\n",a[4]*q,a[4]*q*q,a[4]*q*q*q,a[4]*q*q*q*q,a[4]*q*q*q*q*q);
        }
        else
        {
            printf("None\n");
        }
    }
}