题目大意:
给定一棵n带权树,每个点的权值在[1,n]范围内且互不相等,并满足子结点的权值一定小于父结点。
现在已知一个包含根结点的联通块中个点的权值,求剩下哪些点的权值能够被求出,并求出这些权值。
思路:
贪心。
很显然,对于某一个结点x,如果当前只有一个可取的权值w,且小于其父结点的权值,那么这个结点的权值一定是w。
事实上所有未知结点权值都可以尝试用这样的方法得出,关键是如何唯一确定下这个权值w。
我们可以用一个数组max记录每个结点权值的上界,再用一个数组last记录小于某个权值能取的最大权值。
max数组可以用一趟DFS递归出来。
然后顺序枚举每一个权值,如果可用就加入到一个“黑箱”中,如果现在黑箱中只有一个权值w并且有未知权值的结点,说明这个节点的权值就是w。
然后如果现在黑箱中有多个权值,未知的结点也很多,那么这些权值就作废。
1 #include<cstdio> 2 #include<cctype> 3 #include<vector> 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<int> 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<c[x].size();i++) { 25 dfs(c[x][i]); 26 } 27 } 28 int main() { 29 n=getint(); 30 for(int i=1;i<=n;i++) last[i]=i; 31 for(int i=1;i<=n;i++) { 32 int p=getint(); 33 w[i]=max[i]=getint(); 34 if(p!=i) { 35 c[p].push_back(i); 36 par[i]=p; 37 } else { 38 root=i; 39 w[i]=n; 40 } 41 last[w[i]]=w[i]-1; 42 } 43 dfs(root); 44 for(int i=1,tmp=0;i<=n;i++) { 45 if(last[i]==i) tmp++; 46 if(cnt[i]&&tmp==1) w[s[i]]=i; 47 tmp-=cnt[i]; 48 } 49 for(int i=1;i<=n;i++) printf("%d\n",w[i]); 50 return 0; 51 }