Tree Restoration(猜测一棵树的合法性)

原题:Gym - 101755F

题意:

有n个结点,给出这个结点下面子树的所有结点,问有没有这样的树,如果有,输出所有边

解析:

把给出的条件先记录下来,然后靠这些条件把树建起来,然后bfs这棵树,把bfs的条件和给出的条件对比即可

这样就不用自己去想有哪些bug了

代码:




//从下往上建图(相邻结点),从上往下bfs(只要是上下),检测前后(总关系)是否矛盾

/////
int now=0;
int head[1009],to[10090],nex[10090];

void add(int a,int b){
    to[++now]=b,nex[now]=head[a],head[a]=now;
}
/////
struct node{
    int idx,s;
    bool operator < (const node a)const{
        return s<a.s;
    }
}e[1009];
/////
int f[1009];
int lin[1009][1009],lin2[1009][1009];
/////
void dfs(int p,int now){
    lin2[p][now]=1;
    for(int i=head[now];~i;i=nex[i]){
        dfs(p,to[i]);
    }
}
void out(int p){
    for(int i=head[p];~i;i=nex[i]){
        printf("%d %d\n",p,to[i]);
        out(to[i]);
    }
}
/////
int main(){
    int n=read();
    mmm(head,-1);
    for1(i,1,n){
        e[i].idx=i,e[i].s=read();
        for1(j,1,e[i].s){
            int b=read();
            lin[i][b]=1;
        }
    }
    sort(e+1,e+1+n);
    for1(i,1,n){
        int r=e[i].idx;
        for1(j,1,n){
            if(f[j]==1&&lin[r][j]){
                add(r,j);
                f[j]=0;
            }
        }
        f[r]=1;//可以被上面的点连接
    }
    int yes=1;
    if(now!=n-1)yes=0;//边数不对
    for(int i=1;i<=n;i++){
        dfs(i,i);
        lin2[i][i]=0;
    }

    for1(i,1,n){
        for1(j,1,n){
            if(lin[i][j]!=lin2[i][j])yes=0;
        }
    }
    if(yes){
        printf("YES\n");
        out(e[n].idx);
    }
    else{
        printf("NO\n");
    }
}














猜你喜欢

转载自blog.csdn.net/jk_chen_acmer/article/details/81123117