PKU1659+havel-hakimi定理

http://www.docin.com/p-540438846.html
该定理链接。

/*
havel-hakimi定理
http://www.docin.com/p-540438846.html
*/
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<iostream>
#include<queue>
#include<map>
#include<math.h>
using namespace std;
typedef long long ll;
//typedef __int64 int64;
const int maxn = 24;
const int inf = 0x7fffffff;
const double pi=acos(-1.0);
const double eps = 1e-8;
struct node{
    int num,d;
}a[ maxn ];
int mat[ maxn ][ maxn ];
int cmp( node a,node b ){
    return a.d>b.d;
}
bool flag;

int Judge( int n ){
    for( int i=0;i<n;i++ ){
        if( a[i].d<0 ){
            flag = false;
            return 0;
        }
    }
    for( int i=0;i<n;i++ ){
        if( a[i].d>0 ){
            return 1;
        }
    }
    flag = true;
    return 0;
}
 
int main(){
    int ca;
    scanf("%d",&ca);
    int cc = 0;
    while( ca-- ){
        int n;
        scanf("%d",&n);
        memset( mat,0,sizeof( mat ) );
        for( int i=0;i<n;i++ ){
            scanf("%d",&a[ i ].d);
            a[ i ].num = i;
        }
        if( cc!=0 ){
            printf("\n");
        }
        cc++; 
        flag = true;
        while( 1 ){
            if( Judge( n )==0 ) break;
            sort( a,a+n,cmp );
            for( int i=1;i<=a[0].d;i++ ){
                a[ i ].d--;
                mat[ a[0].num ][ a[ i ].num ] = mat[ a[ i ].num ][ a[ 0 ].num ] = 1;
            }
            a[ 0 ].d = 0;
        //    if( Judge( n )==0 ) break;
        }
        if( flag==false ) printf("NO\n");
        else {
            printf("YES\n");
            for( int i=0;i<n;i++ ){
                for( int j=0;j<n;j++ ){
                    if( j==0 ) printf("%d",mat[i][j]);
                    else printf(" %d",mat[i][j]);
                }
                printf("\n");
            }
        }
    }
    return 0;
}