D1. Constrained Tree 构造+dfs

题目链接:https://codeforces.com/contest/513/problem/D1

题意

你要构造一个 n n n 个结点的树,这棵树按照中序遍历的顺序进行标号(即最后的树用中序遍历应该要是一个 1 − n 1-n 1n 的序列),你现在要满足 m m m 个条件,每个条件的格式是 a a a b b b o p op op 如果 o p op op L E F T LEFT LEFT ,那么 b b b a a a 的左子树上,如果为 R I G H T RIGHT RIGHT 则在右子树上,如果能构造出来,请用前序遍历输出任意一种答案,否则为“不可能”。

做法

先要保证满足几个条件。

  • 在所给出的条件中, a a a 必须要比 b b b 小。
  • 所有子树的左子树上的最大值,必须比右子树上的最小值小。

在满足这两个条件的前提下也不一定能保证合法,但是先开始构造吧。

很神奇的构造方法

① 这里要一直更新一个 l o w low low 值,表示从这个节点往下至少要更新到多少的结点。如果这个节点的左子树上有值,那么直接就往左子树上加上下一个点, l o w low low 值更新为这个左子树上的最大值。如果右子树上有值,那么就把 l o w low low 更新为原最大值和这个右子树所需最大值的max。如果这里的结点更新不到所谓的 l o w low low ,那么就在右子树上继续加点。

② 如果加完某一个左子树后发现,超过了其右子树的最小值,那么也是要判不合法的。

其实很巧妙,上面的 ② 保证了同一个结点上左右子树间的关系,而 ① 则保证了每一个子树下面的最大值是合法的。

代码

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
 
typedef long long ll;
const int maxn=1000005;
int n,c,mir[maxn],mar[maxn],ls[maxn];
int mil[maxn],mal[maxn],cnt,flag,rs[maxn];
void dfs(int u,int low){
    if(flag) return ;
    if(mal[u]){
        ls[u]=++cnt;
        dfs(ls[u],mal[u]);
        mal[u]=0;
        if(cnt>=mir[u]) flag=1;
    }
 
    if(mar[u]){
        rs[u]=++cnt;
        dfs(rs[u],max(low,mar[u]));
        mar[u]=0;
    }
    if(cnt>=low) return ;
    if(!rs[u]) rs[u]=++cnt;
    dfs(rs[u],low);
}
void out(int u){
    if(!u) return ;
    out(ls[u]);
    printf("%d ",u);
    out(rs[u]);
}
int main() {
    scanf("%d%d",&n,&c);
    rep(i,1,n) mil[i]=mir[i]=maxn;
    rep(i,1,c){
        int rt,son; char s[10];
        scanf("%d%d%s",&rt,&son,s+1);
        if(rt>=son) flag=1;
        if(s[1]=='L'){
            mil[rt]=min(mil[rt],son);
            mal[rt]=max(mal[rt],son);
        }
        else{
            mir[rt]=min(mir[rt],son);
            mar[rt]=max(mar[rt],son);
        }
    }
    rep(i,1,n){
        if(mir[i]!=maxn&&mal[i]!=0&&mal[i]>=mir[i]) return 0*printf("IMPOSSIBLE\n");
    }
    cnt=1;
    dfs(1,n);
    if(flag) printf("IMPOSSIBLE\n");
    else {
        out(1);
        printf("\n");
    }
	return 0;
}
©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页