POJ 1659 A - Frogs' Neighborhood

A - Frogs' Neighborhood

Time Limit:5000MS     Memory Limit:10000KB     64bit IO Format:%I64d & %I64u

Submit Status

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 

#include<stdio.h>
#include<string.h>

int g[20][20];
struct node
{
    int data;
    int b;
}a[50];
int main()
{
    int T,n;
    scanf("%d",&T);
    while(T--)
    {
        memset(g,0,sizeof(g));
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i].data);
            a[i].b=i;
        }
        int flag=1;
        while(flag!=0)
        {
            for(int i=0;i<n-1;i++)
            {
                for(int j=i+1;j<n;j++)
                {
                    if(a[i].data < a[j].data)
                    {
                        struct node t=a[i];
                        a[i]=a[j];
                        a[j]=t;
                    }
                }
            }
            if(a[0].data == 0)
                break;
            else
            {
                for(int j=1;j<=a[0].data;j++)
                {
                    a[j].data--;
                    if(a[j].data < 0)
                    {
                        flag = 0;
                        break;
                    }
                    g[a[j].b][a[0].b] = 1;
                    g[a[0].b][a[j].b] = 1;
                }

            }
            a[0].data = 0;
        }
        if(flag != 0)
        {
            printf("YES\n");
            for(int i=0;i<n;i++)
            {
                for(int j=0;j<n;j++)
                {
                    printf("%d",g[i][j]);
                    if(j == n-1)
                    {
                        printf("\n");
                    }
                    else
                    {
                        printf(" ");
                    }
                }
            }
        }
        else
            printf("NO\n");
        printf("\n");


    }
    return 0;
}