HDU 3818 A + B Problem

http://acm.hdu.edu.cn/showproblem.php?pid=3818

Sample Input
2
1 2
2 2 4
3 2 4 6
2 2 5

Sample Output
Case 1:
1 5        //答案的第一个数表示后面跟多少个数
Case 2:
2 5 7

题意：每2行的数用输入斐波那契的表示法，例如上面第一个案例，总和就是sum=f(2)+f(2)+f(4)【第一行：2，第二行：2 4】
要求输出sum的斐波那契表示法，sum=f(2)+f(2)+f(4)=f(0)+f(1)+f(2)+f(4)=f(3)+f(4)=f(5),所以答案是1个数，就是5

再仔细观察，找规律：

①显然地：当b-a==1时，f(a)+f(b)=f(b+1)，当a-b=1时，f(b)+f(a)=f(a+1)

②当a == b时，f(a) + f(b) = f(a-2) + f(a+1);
证明：
f(a) = f(a-2) + f(a-1);
又因为：f(b) = f(a);
所以f(a)+f(b)=f(a-2)+f(a-1)+f(a)=f(a-2)+f(a+1);

#include <iostream>   
#include <fstream>   
#include <algorithm>   
#include <string>   
#include <set>   
//#include <map>   
#include <queue>   
#include <utility>   
#include <stack>   
#include <list>   
#include <vector>   
#include <cstdio>   
#include <cstdlib>   
#include <cstring>   
#include <cmath>   
#include <ctime>   
#include <ctype.h>   
using namespace std;   
#define inf 10000000   //定义无穷大
  
int ans[405];
  
int main()   
{   
    int t, n, i, x, tt, ind, k = 1, j;   
    bool flag;   
    scanf ("%d", &t);
    while (t--)
    {
        tt = 2, ind = 0;
        while (tt--)   
        {   
            scanf ("%d", &n);   
            for (i = 0; i < n; i++)   
            {   
                scanf ("%d", &x);   
                ans[ind++] = x;   //2行输入加入到数组ans
            }   
        }   
        sort (ans, ans+ind);   
        flag = true;
        while (flag)   //不存在可以合并的2个数则跳出
        {   
            flag = false;   
            for (i = 0; i < ind - 1; i++)   
            {   
                if (ans[i] == inf)   
                    continue;   
                for (j = i + 1; j < ind; j++)   
                {   
                    if (ans[j] == inf)
                        break;   
                    if (abs (ans[j] - ans[i]) > 1)
                        continue;   
                    if (!flag)   //说明存在2个数，他们的差绝对值<=1，即可以合并
                        flag = true;   
                    if (ans[j] - ans[i] == 1)   //当b-a==1时，f(a)+f(b)=f(b+1)
                    {
                        ans[ind++] = ans[j] + 1;   //生成"f(b+1)"
                        ans[i] = ans[j] = inf;//把f(a)和f(b)变成无穷大，排序时就可以刷下去了
                    }
                    else if (ans[i] - ans[j] == 1)//当a-b=1时，f(b)+f(a)=f(a+1)，同理
                    {
                        ans[ind++] = ans[i] + 1;
                        ans[i] = ans[j] = inf;
                    }
                    else if (ans[i] == ans[j])//当a == b时，f(a) + f(b) = f(a-2) + f(a+1)
                    {
                        ans[ind++] = ans[i] - 2;   //生成2个
                        ans[ind++] = ans[i] + 1;
                        ans[i] = ans[j] = inf;//相当于删除f(a)和f(b)，实现合并操作
                    }
                }   
            }   
            sort (ans, ans+ind);    //排序，把无穷大刷下去
            /*for (i = 0; i < ind; i++)
                cout << ans[i] << ' ';
            cout << "\n";*/
        }   
        printf ("Case %d:\n", k++);
        int num = 0;   
        for (i = 0; i < ind; i++)   
        {   
            if (ans[i] == inf)   //由于已排序，一旦扫描到无穷大，后面肯定没有合法的数
                break;   
            if (ans[i] > 0)   
                num++;   //累计结果个数，0不算入斐波那契表示法
        }   
        printf ("%d", num);   
        for (i = 0; i < ind; i++)   
        {   
            if (ans[i] == inf)   
                break;   
            if (ans[i] > 0)   
                printf (" %d", ans[i]);   
        }   
        printf ("\n");   
    }   
    return 0;   
}