题目大意:
给定一棵n带权树,每个点的权值在[1,n]范围内且互不相等,并满足子结点的权值一定小于父结点。 现在已知一个包含根结点的联通块中个点的权值,求剩下哪些点的权值能够被求出,并求出这些权值。思路:
贪心。 很显然,对于某一个结点x,如果当前只有一个可取的权值w,且小于其父结点的权值,那么这个结点的权值一定是w。 事实上所有未知结点权值都可以尝试用这样的方法得出,关键是如何唯一确定下这个权值w。 我们可以用一个数组max记录每个结点权值的上界,再用一个数组last记录小于某个权值能取的最大权值。 max数组可以用一趟DFS递归出来。 然后顺序枚举每一个权值,如果可用就加入到一个“黑箱”中,如果现在黑箱中只有一个权值w并且有未知权值的结点,说明这个节点的权值就是w。 然后如果现在黑箱中有多个权值,未知的结点也很多,那么这些权值就作废。1 #include2 #include 3 #include 4 inline int getint() { 5 char ch; 6 while(!isdigit(ch=getchar())); 7 int x=ch^'0'; 8 while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0'); 9 return x;10 }11 const int V=1000001;12 std::vector c[V];13 int n;14 int w[V],s[V],par[V],max[V],cnt[V],last[V],root;15 int find(const int x) {16 return x==last[x]?x:last[x]=find(last[x]);17 }18 void dfs(const int x) {19 if(!max[x]) {20 max[x]=find(max[par[x]]-1);21 s[max[x]]=x;22 cnt[max[x]]++;23 }24 for(unsigned i=0;i