poj 1659 Frogs' Neighborhood

Frogs' Neighborhood 

Description

未名湖附近共有N个大小湖泊L₁, L₂, ..., L_n(其中包括未名湖)，每个湖泊L_i里住着一只青蛙F_i(1 ≤ i ≤ N)。如果湖泊L_i和L_j之间有水路相连，则青蛙F_i和F_j互称为邻居。现在已知每只青蛙的邻居数目x₁, x₂, ..., x_n，请你给出每两个湖泊之间的相连关系。

Input

第一行是测试数据的组数T(0 ≤ T ≤ 20)。每组数据包括两行，第一行是整数N(2 < N < 10)，第二行是N个整数，x₁, x₂,..., x_n(0 ≤ x_i ≤ 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 

分析：题目的意思是说，给你n个顶点，还有n个顶点的度，问你能否构造一个图，如果能，则输出“YES”，并且输出构造的图中的一种，否则输出“NO”。

这个题目是Havel-Hakimi定理的应用，Havel-Hakimi定理是说：由非负整数组成的非增序列s：d1,d2,……,dn是可图的的充要条件是序列s1：d2-1,d3-1,……,d(d1+1)-1,d(d1+2),……,dn是可图的。

根据这个定理，我们可以知道如果一个度的序列是可以构造成图的，那么这个序列的非递增形式从第2个开始到第d1+1个位置减去1是非负的，依次循环，直到第一次出现负数或序列中的每个数都变为0为止。前者是不可图的，后者是可图的。

针对此题，我们还要构建出任意一个图，所以我们可以用一个结构体把图中的顶点的下标记录下来，然后每次减一，就说明相应的位置上有边相连。

下面是我的代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
struct node
{
   int x;//记录顶点的度 
   int r;//记录顶点对应的下标 
};
int n;
node a[15];
int map[15][15];
bool cmp(node a,node b)
{
   if(a.x==b.x) return a.r<b.r;
   return a.x>b.x;
}
int main()
{
   int t;
   scanf("%d",&t);
   while(t--)
   {
      scanf("%d",&n);
      int m=n;
      for(int i=0;i<n;i++)
      {
         scanf("%d",&a[i].x);
         a[i].r=i;
      }
      bool flag=0,b=0;
      
      sort(a,a+n,cmp);
      
      memset(map,0,sizeof(map));
      
      //依次循环，直到出现负数或全为0为止 
      while(1)
      {
         for(int i=0;i<n;i++)
            if(a[i].x<0) {flag=1;break;}
         if(flag) break;
         if(!a[0].x) {b=1;break;}
         for(int i=1;i<=a[0].x;i++)
         {
            a[i].x--;
            map[a[0].r][a[i].r]=map[a[i].r][a[0].r]=1;
         }
         for(int i=0;i<n-1;i++)
            a[i]=a[i+1];
         n--;
         sort(a,a+n,cmp);
      }
      
      if(flag) printf("NO\n\n");
      if(b)
      {
         printf("YES\n");
         for(int i=0;i<m;i++)
         {
            for(int j=0;j<m;j++)
            { 
               printf("%d",map[i][j]);
               if(j<m-1) printf(" ");
            }
            printf("\n");
         }
         printf("\n");
      }
   }
   
   return 0;
}