Havel-Hakimi定理的方法来构图

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef struct {
    int id;
    int d;
}Arg;

Arg arg[10];
int ans[20][20];

int cmp(const void *a,const void *b){
    return ((Arg*)b)->d-((Arg*)a)->d;
}//从大到小 

int main(){
    int t,n;
    cin>>t;
    while(t--){
        cin>>n;
        memset(ans,0,sizeof(ans));//每一次，将数组置0 
        for(int i=0;i<n;i++){
            cin>>arg[i].d;
            arg[i].id=i;
        }
        int k=0,i,j;
        while(k<n){
            qsort(arg+k,n-k,sizeof(arg[0]),cmp);//第一次排序后，每次删除第一个数排序
             if(arg[k].d>n-k-1)
             break;//如果当前度数已经大于后面的长度，这说明不可能是图 
             for(i=1;i<=arg[k].d;i++){//次数 
                
                if(arg[i+k].d<=0)//小于0跳出 
                break;
                arg[i+k].d--;
                ans[arg[k].id][arg[k+i].id]=ans[arg[k+i].id][arg[k].id]=1;//将后一个数的id，指向前一个数 
            }
            if(i<=arg[k].d)//中途跳出 
            break;
            k++;
        }
        if(k<n)//没有删完，就跳出了 
        cout<<"NO"<<endl;
        else
        {
            cout<<"YES"<<endl;
            for(i=0;i<n;i++){
                for( j=0;j<n;j++)
                    cout<<ans[i][j]<<" ";
                    cout<<endl;    
            }
        }
        
    }
    return 0;
}