[2016ACM多校] HDU5764 搜索

题意

给一棵树，结点编号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;
}