GYM 101755 I.Guess the Tree（构造）

Description

已知有一棵高度为$h$的完全二叉树，节点数即为$n=2^h-1$，每次可以查询树上两个点的最短距离，要求在至多$2.5\cdot h\cdot n$次查询内还原该棵树

Input

第一行一整数$n$，保证$n+1$为$2$的整数次幂，每次查询后会输出查询的两个点的最短距离$(1\le n\le 1023)$

Output

每次查询输出$?\ u\ v$表示查询树上两点$u,v$之间的最短距离，还原好该棵树后输出$!\ fa[1]\ ...\ fa[n]$，其中$fa[i]$表示$i$节点的父亲节点，定义根节点的父亲节点为$0$

Sample Input

3

1

1

Sample Output

? 1 2

? 2 3

! 2 0 2

Solution

首先固定一个点，然后查询$n-1$次表示其他$n-1$个点到该点的距离，其中最远点$x$必然为一个叶子节点，然后查询其他点到$x$的距离$dis1[i]$，如果某个$dis1[y]=2h-2$，说明$y$与$x$是根节点的两棵不同子树的叶子节点，然后查询所有节点到$y$的距离$dis2[i]$，如果$dis1[i]=dis2[i]$说明$i$是根节点，如果$dis1[i]<dis2[i]$说明$i$节点和$x$在同一棵子树，否则说明$i$节点和$y$在同一棵子树，以此可以把点分成根节点，左子树和右子树然后递归下去即可，注意到第一次计算时，只要$dis1[i]=2h-2$或$dis2[i]=2h-2$即说明$i$节点是叶子节点，以此标记所有叶子节点，那么之后的递归中就不用查询$n-1$次找叶子了，总查询数$2(h-1)n+n$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1025;
int n,a[maxn],fa[maxn],vis[maxn],dis1[maxn],dis2[maxn];
vector<int>v1,v2;
int Solve(int l,int r,int h)
{
    if(l==r)return a[l];
    int x,y,root;
    if((1<<h)==n+1)
    {
        int dis=0,temp;
        for(int i=l+1;i<=r;i++)
        {
            printf("? %d %d\n",a[l],a[i]);
            cout.flush();
            scanf("%d",&temp);
            if(temp>dis)dis=temp,x=a[i];
        }
        vis[x]=1;
    }
    else 
    {
        for(int i=l;i<=r;i++)
            if(vis[a[i]])
            {
                x=a[i];
                break;
            }
    }
    for(int i=l;i<=r;i++)
        if(a[i]!=x)
        {
            printf("? %d %d\n",a[i],x);
            cout.flush();
            scanf("%d",&dis1[a[i]]);
            if(dis1[a[i]]==2*h-2)y=a[i];
        }
    v1.clear();v2.clear();
    v1.push_back(x);
    v2.push_back(y);
    for(int i=l;i<=r;i++)
        if(a[i]!=x&&a[i]!=y)
        {
            printf("? %d %d\n",a[i],y);
            cout.flush();
            scanf("%d",&dis2[a[i]]);
            if(dis1[a[i]]==dis2[a[i]])root=a[i];
            else if(dis1[a[i]]<dis2[a[i]])v1.push_back(a[i]);
            else v2.push_back(a[i]);
        }
    if((1<<h)==n+1)
    {
        vis[y]=1;
        for(int i=l;i<=r;i++)
            if(a[i]!=x&&a[i]!=y&&(dis1[a[i]]==2*h-2||dis2[a[i]]==2*h-2))
                vis[a[i]]=1;
    }
    int mid=l+v1.size()-1;
    for(int i=0,j=l;i<v1.size();i++,j++)a[j]=v1[i];
    for(int i=0,j=mid+1;i<v2.size();i++,j++)a[j]=v2[i];
    fa[Solve(l,mid,h-1)]=root;
    fa[Solve(mid+1,r-1,h-1)]=root;
    return root;
}
int main()
{
    scanf("%d",&n);
    if(n==1)printf("! 0\n");
    else
    {
        for(int i=1;i<=n;i++)a[i]=i;
        int h=1;
        while((1<<h)<n+1)h++;
        fa[Solve(1,n,h)]=0;
        printf("! ");
        for(int i=1;i<=n;i++)printf("%d%c",fa[i],i==n?'\n':' ');
    }
    return 0;
}