poj 1659 A - Frogs' Neighborhood

问题描述

未名湖附近共有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。每两个数字之间输出一个空格。如果存在多种可能，只需给出一种符合条件的情形。相邻两组测试数据之间输出一个空行。

样例输入

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

样例输出

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定理判断是否能构成一个图，然后再每次变化的时候，记录两个点的坐标，并且都赋值成1即可；

代码如下：

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
using namespace std;

const int maxx=15;
int n;
int map[maxx][maxx];

struct node
{
    int du;int id;
}no[maxx];

bool cmp(node a,node b)
{
    return a.du>b.du;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        memset(map,0,sizeof(map));
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&no[i].du);
            no[i].id=i;
        }
        bool mark=true;

        while(1)
        {
            sort(no+1,no+1+n,cmp);
            if(no[1].du==0)
            {
                mark=true;break;
            }

            for(int i=1;i<=no[1].du;i++)
            {
                no[i+1].du--;
                if(no[i+1].du<0)
                {
                     mark=false;break;
                }
                map[no[1].id][no[1+i].id]=1;
                map[no[1+i].id][no[1].id]=1;
            }
            no[1].du=0;
            if(!mark)
                break;
        }
            if(mark)
            {
                printf("YES\n");
                for(int i=1;i<=n;i++)
                {
                    for(int j=1;j<=n;j++)
                   cout<<map[i][j]<<" ";
                    cout<<endl;
                }

            }
            else
                printf("NO\n");
                if(t)cout<<endl;
    }
}