POJ1659Frogs' Neighborhood(lavel定理)

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

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

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

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
struct node
{
    int indx,ans;
}a[15];
int cmp(node a,node b)
{
    return a.ans>b.ans;
}
int main()
{
    int map[15][15],t,n,flag=0;
    scanf("%d",&t);
    while(t--)
    {
        if(flag)printf("\n");flag=1;
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i].ans); a[i].indx=i;
        }
        int tn=n;
        memset(map,0,sizeof(map));
        while(n)
        {
            sort(a,a+n,cmp);
            int i;
            for(i=0;i<n;i++)
            if(a[i].ans==0)
            {
                n=i; break;
            }
            i=1;
            while(a[0].ans&&i<n&&a[i].ans)
            {
                a[0].ans--; a[i].ans--;
                map[a[0].indx][a[i].indx]=map[a[i].indx][a[0].indx]=1;
                i++;
            }
            if(a[0].ans>0)break;
        }
        if(n)
        printf("NO\n");
        else
        {
            printf("YES\n");
            for(int i=0;i<tn;i++)
            {
                for(int j=0;j<tn;j++)
                if(j==0) printf("%d",map[i][j]);
                else printf(" %d",map[i][j]);
                printf("\n");
            }
        }
    }
}