题意
给一棵树,结点编号1~N,权值原本是1~N的一个排列。现在让每个点的权值更新为它所在子树的最大权值。给出更新后的树,输出所满足的最小字典序的1~N原排列,否则输出不可能。(根不给定)
思路
赛场上没看到最小字典序,榜也有点歪去坑03了,读题很重要!
更新就是回溯是当前结点变为自己和上传的最大值并上传。显然,所有的值一定是一个链不能中断也不能分叉,根一定是值为n的链的端点。这样根最多只有两个情况,其实这两个情况真值相同,且让小标号端点为根一定更优。因为不论n这个最大值从哪个方向传递,此链之外的部分都不受影响维持原来的状态,反正也不可能更新这个顶端链,所以最小字典序时当然要把n放到大标号的点去了。
既然找到了根就好办了,先把1~N放进一个集合,然后每次遇到一个点的子节点都比它小就取出他现在的值,记录为他的原始值,并标记。如果这个值已经被用掉了那就不可能还原。然后从n->1枚举每个没有确定值的点,标记为比他上线小的第一个没被使用的值就得到了最小字典序的解。如果剩余值不够用了那就不可能。
AC代码 C++
#include <stdio.h>
#include <string.h>
#include <vector>
#include <set>
using namespace std;
#define MAXN 100005
vector<int> G[MAXN];
int value[MAXN];
bool sure[MAXN];
set<int> res;
int dfs(int u, int fa)
{
int i, v, next=u;
for(i=G[u].size(); i--;)
if((v = G[u][i]) != fa)
{
if(value[v] == value[u])
{
if(next != u)
return -1;
next = v;
}
if(value[v] > value[u] || dfs(v, u))
return -1;
}
if(next == u)
if(res.find(value[u]) != res.end())
{
res.erase(value[u]);
sure[u] = true;
}
else
return -1;
return 0;
}
int main()
{
int T, t, n, i, root, a, b;
scanf("%d", &T);
for(t=1; t<=T; t++)
{
scanf("%d", &n);
for(i=1, res.clear(); i<=n; i++)
{
scanf("%d", value+i);
G[i].clear();
res.insert(i);
}
for(i=1; i<n; i++)
{
scanf("%d%d", &a, &b);
G[a].push_back(b);
G[b].push_back(a);
}
for(i=1, root=0; i<=n && !root; i++)
if(value[i] == n)
{
for(a=G[i].size(), b=0; a--;)
if(value[G[i][a]] == n)
b++;
if(b <= 1)
root = i;
}
memset(sure, false, sizeof sure);
printf("Case #%d: ", t);
if(!root || dfs(root, 0))
goto next;
for(i=n+1; --i;)
if(!sure[i])
{
set<int>::iterator it = res.lower_bound(value[i]);
if(it != res.begin())
{
value[i] = *--it;
res.erase(it);
}
else
goto next;
}
for(i=1; i<=n; i++)
printf("%d%c", value[i], i!=n?' ':'\n');
continue;
next: puts("Impossible");
}
return 0;
}