华为机考真题 -- 剩余银饰的重量

题目描述：

有 N 块二手市场收集的银饰，每块银饰的重量都是正整数，收集到的银饰会被熔化用于打造新的饰品。

每一回合，从中选出三块最重的银饰，然后一起熔掉。假设银饰的重量分别为 x 、y 和 z，且 x <= y <= z。那么熔掉的可能结果如下：
1. 如果 x == y == z，那么三块银饰都会被完全熔掉；
2. 如果 x == y 且 y != z，会剩余重量为 z - y 的银块无法被熔掉；
3. 如果 x != y 且 y == z，会剩余重量为 y - x 的银块无法被熔掉；
4. 如果 x != y 且 y != z，会剩余重量为 z - y 与 y - x 差值的银块无法被熔掉。
5. 最后，如果剩余两块，返回较大的重量（若两块重量相同，返回任意一块皆可）；如果只剩
下一块，返回该块的重量；如果没有剩下，就返回 0。

输入描述：

输入数据为两行;
第一行为银饰数组长度 n，1 ≤ n ≤ 40;
第二行为 n 块银饰的重量，重量的取值范围为[1，2000]，重量之间使用空格隔开;

输出描述：

如果剩余两块，返回较大的重量（若两块重量相同，返回任意一块皆可）；如果只剩下一块，返回该块的重量；如果没有剩下，就返回 0。

示例1:

输入
3
1 1 1

输出
0
说明：选出 1 1 1，得到 0，最终数组转换为 []，最后没有剩下银块，返回 0

示例2:

输入
3
3 7 10

输出
1

说明：选出 3 7 10，需要计算 (7-3) 和 (10-7) 的差值，即(7-3)-(10-7)=1，所以数组转换为 [1]，剩余一块，返回该块重量，返回 1

C++源码：

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdlib>
//#include <math>

using namespace std;

int main() {
    int n;
    cin >> n;
    vector<int> weights(n);
    for (int i = 0; i < n; ++i) {
        cin >> weights[i];
        if (cin.get() == '\n') {
            break;
        }
    }

    sort(weights.begin(), weights.end());

    while (weights.size() >=3)
    {
        int x = 0;
        int y = 0;
        int z = 0;
        int len = weights.size();

        cout << "len=" << len << endl;
        z = weights[len - 1];
        y = weights[len - 2];
        x = weights[len - 3];

        if ((x == y) && (y == z)) {
            weights.pop_back();
            weights.pop_back();
            weights.pop_back();
        }
        else if ((x == y) && (y != z)) {
            int temp1 = z - y;
            weights.pop_back();
            weights.pop_back();
            weights.pop_back();
            weights.push_back(temp1);
        }
        else if ((y == z) && (x != y)) {
            int temp2 = z - y;
            weights.pop_back();
            weights.pop_back();
            weights.pop_back();
            weights.push_back(temp2);
        }
        else if ((y != z) && (x != y)) {
            int temp3 = abs(z - y - (y - x));
            weights.pop_back();
            weights.pop_back();
            weights.pop_back();
            weights.push_back(temp3);
        }

        sort(weights.begin(), weights.end());
        
    }

    if (weights.empty()) {
        cout << 0 << endl;
    }
    else if (weights.size() == 1) {
        cout << weights[0] << endl;
    }
    else if (weights.size() == 2) {
        int maxWeights = max(weights[0], weights[1]);
        cout << maxWeights << endl;   
    }

    system("pause");
    return 0;
}