看到不超过 (36) 次询问显然可以想到是 $log $ 时间复杂度。
暴力的做法自然是 (u) 从 (1) 开始,每次询问 (u o x) 路径上的第 (2) 个节点,暴力跳并继续询问,期望询问次数为 (log n),但链的询问次数为 (O(n))。这和树剖很像,都是在链时存在 ( ext{worst case}) ,在期望情况下为 (log n)。于是我们可以考虑树剖优化。
可以先使用一次 ( ext{d 1}) 的询问,得到 (x) 的深度。然后从点 (1) 开始向下跳重链,跳到重链尾再询问 ( ext{d u}),根据回答的值,我们可以算出此时的 ( ext{lca}(u,x))。然后再询问 ( ext{s }lca(u,x)) 得到 (x) 所在的子树位于 ( ext{lca}(u,x)) 的位置,然后继续跳。
但是这样会 ( ext{TLE}),由于每次跳到重链尾都没有保存任何信息,仅凭深度,我们很难得到 ( ext{lca}(u,x)) 的值。所以我们不真正的一次跳到重链尾,而是每次直接跳一步,要么跳到重链尾,要么跳到和 (x) 点深度相同的位置,此时深度小于 (u) 的所有重链上的点信息都已被保存,所以我们很容易得到 ( ext{lca}(u,x)) 的值。
时间复杂度为 (O(n)),能在 (log n) 个询问内得到答案。
Show the Code
#include<cstdio>
#include<cstdlib>
int depx,res,cnt=0;
int h[200005],to[400005],ver[400005];
int mp[200005],size[200005],son[200005],dep[200005],maxDep[200005];
inline int read() {
register int x=0,f=1;register char s=getchar();
while(s>'9'||s<'0') {if(s=='-') f=-1;s=getchar();}
while(s>='0'&&s<='9') {x=x*10+s-'0';s=getchar();}
return x*f;
}
inline int max(const int &x,const int &y) {return x>y? x:y;}
inline void add(int x,int y) {to[++cnt]=y;ver[cnt]=h[x];h[x]=cnt;}
inline void query(int op,int x) {
if(op==1) {printf("d %d
",x);fflush(stdout);scanf("%d",&res);}
else if(op==2) {printf("s %d
",x);fflush(stdout);scanf("%d",&res);}
else {printf("! %d
",x);exit(0);}
}
inline void dfs(int x,int fa) {
size[x]=1; maxDep[x]=dep[x];
for(register int i=h[x];i;i=ver[i]) {
int y=to[i];
if(y==fa) continue;
dep[y]=dep[x]+1; dfs(y,x); size[x]+=size[y]; maxDep[x]=max(maxDep[x],maxDep[y]);
if(size[son[x]]<size[y]&&maxDep[y]>=depx) son[x]=y;
}
}
inline void solve() {
int cur=1;
while(1) {
// printf("%d
",cur);
if(dep[cur]==depx) {
query(1,cur);
if(res==0) {query(3,cur);}
cur=mp[depx-res/2];
query(2,cur);
if(dep[res]==depx) {query(3,res);}
cur=res;
}
mp[dep[cur]]=cur;
cur=son[cur];
}
}
int main() {
int n=read();
for(register int i=1;i<n;++i) {int x=read(),y=read();add(x,y);add(y,x);}
query(1,1); depx=res+1;
dep[1]=1; dfs(1,-1);
solve();
return 0;
}