poj 1659 Frogs' Neighborhood

题目链接：http://poj.org/problem?id=1659

知识扩展：（Havel-Hakimi 定理）由非负整数组成的非增序列 s: d₁, d₂, ··· , d_n (n≥2, d≥1) 是可图的，当且仅当序列 s₁ : d₂-1, d₃-1, ··· , d_d1+1-1, d_d1+2, ··· , d_n 是可图的。序列 s₁ 中有 n-1 个非负整数，s 序列中 d₁ 后的前 d₁ 个度数（即 d₂~d_d1+1）减 1 后构成 s₁ 中的前 d₁ 个数。

#include<iostream>
#include<algorithm>
using namespace std;

struct Edge{
    int id,deg;
    bool operator <(const Edge& E) const{
        return deg>E.deg;
    }
}e[10];

int main(){
    int t,n,i,j,a[10][10];
    cin>>t;
    while(t--){
        cin>>n;
        for(i=0;i<n;i++){
            for(j=0;j<n;j++) a[i][j]=0;
        }
        for(i=0;i<n;i++){
            cin>>e[i].deg;
            e[i].id=i;
        }
        bool flag=true;
        i=0;
        while(flag){
            sort(e+i,e+n);
            if(e[i].deg==0) break;
            for(j=i+1;j<=i+e[i].deg;j++){
                if(e[j].deg-1>=0){
                    a[e[i].id][e[j].id]=a[e[j].id][e[i].id]=1;
                    e[j].deg--;
                }
                else{
                    flag=false;
                    break;
                }
            }
            i++;
        }
        if(!flag) cout<<"NO"<<endl;
        else{
            cout<<"YES"<<endl;
            for(i=0;i<n;i++){
                for(j=0;j<n-1;j++) cout<<a[i][j]<<" ";
                cout<<a[i][n-1]<<endl;
            }    
        }
        if(t>0) cout<<endl;
    }
    return 0;
}

---