Frogs‘ Neighborhood(POJ 1659 CC++)

poj 1659

Description

未名湖附近共有N个大小湖泊L1, L2, …, Ln(其中包括未名湖)，每个湖泊Li里住着一只青蛙Fi(1 ≤ i ≤ N)。如果湖泊Li和Lj之间有水路相连，则青蛙Fi和Fj互称为邻居。现在已知每只青蛙的邻居数目x1, x2, …, xn，请你给出每两个湖泊之间的相连关系。

Input

第一行是测试数据的组数T(0 ≤ T ≤ 20)。每组数据包括两行，第一行是整数N(2 < N < 10)，第二行是N个整数，x1, x2,…, xn(0 ≤ xi ≤ N)。

Output

对输入的每组测试数据，如果不存在可能的相连关系，输出"NO"。否则输出"YES"，并用N×N的矩阵表示湖泊间的相邻关系，即如果湖泊i与湖泊j之间有水路相连，则第i行的第j个数字为1，否则为0。每两个数字之间输出一个空格。如果存在多种可能，只需给出一种符合条件的情形。相邻两组测试数据之间输出一个空行。

Sample Input

3 7 4 3 1 5 4 2 1 6 4 3 1 4 2 0 6 2 3 1 1 2 1

Sample Output

YES 0 1 0 1 1 0 1 1 0 0 1 1 0 0 0 0 0 1 0 0 0 1 1 1 0 1 1 0 1 1 0 1 0 1 0 0 0 0 1 1 0 0 1 0 0 0 0 0 0 NO YES 0 1 0 0 1 0 1 0 0 1 1 0 0 0 0 0 0 1 0 1 0 0 0 0 1 1 0 0 0 0 0 0 1 0 0 0

提示：Havel-Hakimi定理

#include <iostream> #include <cstdio> #include <algorithm> #include <memory.h> using namespace std; const int maxn = 12; class Pair { public: int m_id; int val; }; bool cmp(Pair a, Pair b) { return a.val > b.val; } int main() { Pair p[maxn]; int ans[maxn][maxn]; int T; scanf("%d", &T); while (T--) { int n; scanf("%d", &n); for (int i = 0; i < n; i++) { p[i].m_id = i + 1; scanf("%d", &p[i].val); } memset(ans, 0, sizeof(ans)); bool flag = true; int num = n; while (num--) { sort(p, p + n, cmp); if (p[0].val == 0) break; for (int i = 1; i <= p[0].val; i++) { if (p[i].val > 0 && i < n) { ans[p[0].m_id][p[i].m_id] = ans[p[i].m_id][p[0].m_id] = 1; p[i].val--; } else { flag = false; break; } } if (!flag) break; p[0].val = 0; } if (flag) { puts("YES"); for (int i = 1; i <= n; i++) { for (int j = 1; j <= n; j++) { if (j == 1) printf("%d", ans[i][j]); else printf(" %d", ans[i][j]); } putchar('\n'); } } else puts("NO"); if (T > 0) putchar('\n'); } return 0; }