UVa Problem 10249 The Grand Dinner （丰盛的晚餐）

// The Grand Dinner （丰盛的晚餐） // PC/UVa IDs: 111007/10249, Popularity: C, Success rate: high Level: 4 // Verdict: Accepted // Submission Date: 2011-10-08 // UVa Run Time: 0.092s // // 版权所有（C）2011，邱秋。metaphysis # yeah dot net // // [Problem Description] // Each team participating in this year’s ACM World Finals is expected to attend // the grand banquet arranged for after the award ceremony. To maximize the amount // of interaction between members of different teams, no two members of the same // team will be allowed to sit at the same table. // // Given the number of members on each team (including contestants, coaches, // reserves, and guests) and the seating capacity of each table, determine whether // it is possible for the teams to sit as described. If such an arrangement is // possible, output one such seating assignment. If there are multiple possible // arrangements, any one is acceptable. // // [Input] // The input file may contain multiple test cases. The first line of each test // case contains two integers, 1 ≤ M ≤ 70 and 1 ≤ N ≤ 50, denoting the number of // teams and tables, respectively. The second line of each test case contains M // integers, where the ith integer mi indicates the number of members of team i. // There are at most 100 members of any team. The third line contains N integers, // where the jth integer nj , 2 ≤ nj ≤ 100, indicates the seating capacity of // table j. // // A test case containing two zeros for M and N terminates the input. // [Output] // For each test case, print a line containing either 1 or 0, denoting whether // there exists a valid seating arrangement of the team members. In case of a // successful arrangement, print M additional lines where the ith line contains // a table number (from 1 to N ) for each of the members of team i. // // [Sample Input] // 4 5 // 4 5 3 5 // 3 5 2 6 4 // 4 5 // 4 5 3 5 // 3 5 2 6 3 // 0 // // [Sample Output] // 1 // 1 2 4 5 // 1 2 3 4 5 // 2 4 5 // 1 2 3 4 5 // // [解题方法] // 此题有两种解法，一种是简单的贪心法，一种是较复杂的图论网络最大流法。 // // 贪心法采用如下策略：先将人数按降序排列，从编号为 1 的桌子开始安排座位，安排完后，安排下一队。若 // 能安排完所有人，则输出 1 和安排方案，若不能安排则输出 0，其中明显不能安排的，如某队人数大于桌子 // 数，参赛队伍数或桌子数为 0 等特殊情况，可以预先处理（虽然可以得到正确答案，但是贪心法的正确性如 // 何证明？可以通过网络最大流来证明吗？）。 // // 网络流解法：源点 source 和每支参赛队伍之间边的容量为参赛队伍人数，每支队伍到桌子之间边的容量为 // 1，每张桌子到汇点 sink 边的容量为桌子的座位数，然后使用网络流算法求最大流，如果最大流等于参赛队 // 伍总人数，则满足条件，输出方案，否则不满足条件，输出 0。可以使用宽度优先遍历的 Ford-Fullerson // 增广路方法，又名 Edmonds-Karp 算法，算法效率为 O（V*E*E），后续博文会给出。 #include <iostream> #include <algorithm> #include <cstring> using namespace std; #define MAXTEAMS 70 // 最大参赛队伍数。 #define MAXTABLES 50 // 桌子最大数。 #define UNSOLVABLE 0 #define SOLVABLE 1 struct team { int number; // 参赛队伍编号。 int nmembers; // 参赛队伍人数。 }; // 比较函数，参数队伍人数多的排前面。 bool cmp(team x, team y) { return x.nmembers > y.nmembers; } int main(int ac, char *av[]) { team teams[MAXTEAMS]; // 各参赛队编号，人数。 int capacity[MAXTABLES]; // 桌子的座位数。 int nteams, ntables, ntemp; // 队伍数，桌子数，总人数。 bool seat[MAXTABLES][MAXTEAMS]; // 记录座位安排方案。 while (cin >> nteams >> ntables, nteams || ntables) { // 读入参赛队人数并找参赛队的最大人数。 int maxMembers = 0; for (int i = 0; i < nteams; i++) { cin >> ntemp; if (maxMembers < ntemp) maxMembers = ntemp; teams[i].number = i; teams[i].nmembers = ntemp; } // 读入桌子座位数量。 for (int i = 0; i < ntables; i++) cin >> capacity[i]; // 若参赛队伍数为 0，则直接输出存在，但是不用输出具体方案，因为所有桌子无人坐。 if (nteams == 0) { cout << SOLVABLE << "\n"; continue; } // 若桌子数为 0 或者参赛队伍中某队人数超过桌子数，则无法安排。 if (ntables == 0 || maxMembers > ntables) { cout << UNSOLVABLE << "\n"; continue; } // 将参赛队伍按人数多少从大到小排列。 sort(teams, teams + nteams, cmp); // 贪心法安排座位，若某队无法安排，则无安排方案。 bool sitting = true; memset(seat, false, sizeof(seat)); for (int i = 0; i < nteams; i++) { ntemp = teams[i].nmembers; for (int j = 0; j < ntables && ntemp; j++) if (capacity[j]) { ntemp--; capacity[j]--; seat[j][teams[i].number] = true; } // 若经过一轮安排后，剩余人数不为 0 则无法安排。 if (ntemp) { sitting = false; break; } } cout << (sitting ? SOLVABLE : UNSOLVABLE) << "\n"; if (sitting) { for (int i = 0; i < nteams; i++) { int blank = 0; for (int j = 0; j < ntables; j++) if (seat[j][i]) cout << (blank++ ? " " : "") << (j + 1); cout << "\n"; } } } return 0; }