UVa Problem 10032 Tug of War （拔河）

// Tug of War （拔河） // PC/UVa IDs: 110805/10032, Popularity: B, Success rate: low Level: 2 // Verdict: UVa Online Judge - Wrong answer, Programming-Challenges - Solved // Submission Date: 2011-08-09 // Programming-Challenges Run Time: 0.020s // // 版权所有（C）2011，邱秋。metaphysis # yeah dot net // // 直接枚举子集不可行，因为给定的子集数目太大。解题思路如下：确定所有子集所能得到的不同的总重量， // 子集的不同总重量个数要比子集数目少得多。为了获得所有可能的总重量，可以先将重量按从小到大的顺序 // 排列，设第一队人员总数为 i，第二队人员总数为 j，所有人员的总体重为 G，总人数为 N，根据题意， // 两队人员相差不能超过一个人，故 i 和 j 可根据总人数 N 来确定，若 N 为偶数，则 i = j = // [N / 2]，若 N 为奇数，则 i = [N / 2]，j = [N / 2] + 1，符号 [] 表示取整。由于已经将体 // 重值按从小到大的顺序排列，则可以知道前 i 个人的体重和是所能得到的最小体重值，后 j 个人的体重 // 和是能得到的最大体重值。则某队的体重值只可能在此范围内变化，为了得到差值最小的两队体重分布，可 // 以比较前 i 个人与后 j 个人的体重差值，取最先得到最小差值的两个人，交换他们的体重，判断是否达 // 到总体重的一半，若达到可返回，若仍小于总体重的一半，将交换后的两队人员体重值按从小到大的顺序排 // 序，继续寻找体重差值最小的两个人，交换体重，直到前一队人员总体重达到或大于总体重的一半，获取体 // 重和相差较小的一组值。这样可以保证取到所有可能的体重值，在小于 G / 2 和大于 G / 2 的时候需要 // 判断哪个差值更小。 // // 有意思的是，别人使用动态规划算法得到的解法虽然在 Programming Challenges 和 UVa 获得 AC， // 但是某些测试数据却无法通过，我使用这个算法，通过了 UVa 论坛上的所有数据，而且一些 AC 程序都 // 错误的数据也通过了。可见 PC 和 UVa 上的测试数据并不是很强。使用动态规划解题的程序示例可参考 // http://tausiq.wordpress.com/2011/03/07/uva-10032-tug-of-war/ ，评论里有不能通过 // 的测试数据。 // // 在 UVa 上的另外一个问题（题号 562）和此题非常类似，是分硬币的问题，不过和本题不同，分成两部 // 分的硬币没有数目最多相差 1 枚的限制。更一般的这类问题称为平衡划分问题，可以使用动态规划解决，可 // 参考： http://people.csail.mit.edu/bdean/6.046/dp/ 。 // // 本题解法在 Progrmming Challenges 上通过，在 UVa Online Judge 上为 WA。 #include <iostream> #include <vector> #include <algorithm> using namespace std; #define MAX_DIFF 450 // 两个人体重差值最大可能值。 int tug_of_war(vector <int> &weight, int sum) { // 若只有 1 人，显然一队体重为 0，另一队体重即为唯一参加野餐的人的体重。 if (weight.size() == 1) return 0; // 将体重值按增序排列，以便获得总体重值的一个递增序列。 sort(weight.begin(), weight.end()); int lower, upper; lower = weight.size() / 2 - 1; upper = lower + 1; // 查找并替换最先找到的两队中体重值相差最小的两个队员，当前一队的体重总重量小于后一队 // 时，继续该过程，直到前一队体重值等于或大于总体重的一半。 int origin_lower, origin_upper; while (true) { int memo_lower, memo_upper; int diff_weight = MAX_DIFF; // 寻找最先出现的体重差值最小的两个队员。 bool changed = false; for (int m = 0; m <= lower; m++) { for (int n = upper; n < weight.size(); n++) { int tmp = weight.at(n) - weight.at(m); // 保证交换后大于原队伍的体重和。 if (tmp > 0 && tmp < diff_weight) { memo_lower = m; memo_upper = n; diff_weight = weight.at(n) - weight.at(m); changed = true; } } } // 若找到，交换最先出现的差值最小的两个队员。 if (changed == true) { origin_lower = weight.at(memo_lower); origin_upper = weight.at(memo_upper); int tmp = weight.at(memo_lower); weight.at(memo_lower) = weight.at(memo_upper); weight.at(memo_upper) = tmp; } // 计算前一队的体重和。 int team_weight = 0; for (int i = 0; i <= lower; i++) team_weight += weight.at(i); // 未找到差值大于 0 的两个队员，或者前一队体重和等于总体重的一半，返回当前 // 前一队的体重和即为题目所求。 if (changed == false || team_weight == (sum / 2)) return team_weight; // 前一队体重和小于总体重的一半，将两队人员体重各自按增序排列后继续替换。 if (team_weight < (sum / 2)) { sort(weight.begin(), weight.begin() + lower); sort(weight.begin() + upper, weight.end()); continue; } // 前一队人员体重和大于总体重的一半，比较前一次替换和当前两队人员体重差值，取 // 差值较小的一组。 int team_a = 0, team_b = 0; for (int i = 0; i <= lower; i++) team_a += weight.at(i); team_b = team_a + origin_lower - origin_upper; if (abs(2 * team_a - sum) < abs(2 * team_b - sum)) return (sum - team_a); else return team_b; } } int main(int ac, char *av[]) { int cases; // 测试数据组数。 vector <int> weight; // 记录体重值。 cin >> cases; while (cases--) { int n; // 参加野餐总人数。 int tmp; // tmp 为单个人的体重 int sum = 0; // sum 为总的体重。 weight.clear(); cin >> n; while (n--) { cin >> tmp; sum += tmp; weight.push_back(tmp); } // 获取满足题意的体重较小的一队总体重值。 int smaller = tug_of_war(weight, sum); cout << smaller << " " << (sum - smaller) << endl; if (cases) cout << endl; } return 0; }