某易笔试题之搬砖简单解法

题目内容：

    给定n块砖，每块砖有不同的厚度，希望用这些砖堆两个高度相同的塔，使堆出来的高度尽量高，能否实现呢？

输入

输入包括两行：

第一行为一个整数n(0<=n<=50)，一共有n块砖

第二行为n个整数，代表每块砖的厚度h[i](1<=h[i]<=500000)

输出

如果可以堆出两座相同高度的塔，输出最高能拼凑的高度，如果不能则输出-1

一种可行的解（实际上只是可行解，复杂度太高了）

　　将输入排序，得到两组和差不多大的数，然后做差，寻找能弥补差的数，找不到则结束，判断输出是否相等即可。

/*
给定<50块砖，不同高度，堆尽可能高尽可能接近高度的两座塔
输入文件两行：第一行： 砖数，第二行：砖厚0 砖厚1 ...
*/
#include <stdio.h>

void sort(int *a,int n)
{
    int i,j,k,temp;

    for(i=0;i<n-1;i++)
    {
        k=i; /*给记号赋值*/
        for(j=i+1;j<n;j++)
            if(a[k]>a[j ]) k=j; /*是k总是指向最小元素*/
        if(i!=k)/*当k!=i是才交换，否则a[i ] 即为最小*/
        {
            temp=a[i ];
            a[i ]=a[k];
            a[k]=temp;
        }
    }
}

int sum(int *a, int n)
{
    int i;
    int temp = 0;
    for (i = 0; i<n; i++)
    {
        temp += a[i];
    }

    return temp;
}

void main()
{
    int i;
    int n;
    int a[50];

    //freopen("test.txt","r",stdin);    //for debug local
    scanf("%d", &n);
    for (i = 0; i<n; i++)
    {
        scanf("%d", &a[i]);
    }

    if (sum(a, n) % 2 != 0)
        printf("-1");

    sort(a, n);    //升序排序

    int b[50], c[50];
    int len_b = n/2, len_c = n/2;
    for (i = 0; i<n/2; i++)    //分成两堆尽可能接近的砖
    {
        b[i] = a[2*i];
        c[i] = a[2*i+1];
    }
    if (n%2 != 0)    //奇数个，最后一个加入b
    {
        b[i] = a[n-1];
        len_b++;
    }

    int sum_a, sum_b, diff, temp;
    int find = 1, j = 0;
    int abigb = 0;    //1:b>c,0:b<c
    while(find)
    {
        sum_a = sum(b, len_b);
        sum_b = sum(c, len_c);
        abigb = (sum_a > sum_b)?1:0;
        diff = abigb?(sum_a-sum_b):(sum_b-sum_a);
        find  = 0;
        if (abigb)
        {
            temp = 0;
            //找到小于diff的最大数
            for (i = 0; i<len_b; i++)
            {
                if (b[i] < diff && b[i] > temp)
                {
                    temp = b[i];
                    j = i;
                    find  = 1;
                }
            }
            //找到
            if (find == 1)
            {
                c[len_c] = b[j];len_c++;
                b[j] = b[len_b - 1];len_b--;
            }
        }
        else
        {
            temp = 0;
            //找到小于diff的最大数
            for (i = 0; i<len_c; i++)
            {
                if (c[i] < diff && c[i] > temp)
                {
                    temp = c[i];
                    j = i;
                    find  = 1;
                }
            }
            //找到
            if (find == 1)
            {
                b[len_b] = c[j];len_b++;        //搬移最小的
                c[j] = c[len_c - 1];len_c--;
            }
        }
    }

    if (sum_a == sum_b)
        printf("%d", sum_a);
    else
        printf("-1");
}