添加链接描述
通过map记录下标 进行遍历,可以直接得到pre的根
这样就不需要k++,在建树的过程中寻找如果下标都在根两边说明找到了最近的点,否则就往两边找
#includeusing namespace std;const int N=1e4+9;int in[N],pre[N];int n,q;int ok=0;unordered_mappt;void lca(int inl,int inr,int root,int a,int b){ if(ok)return; if(inrk&&ptb>k){ lca(k+1,inr,root+(k-inl)+1,a,b); } else if((ptak)||(pta>k&&ptb>q>>n; for(int i=1;i<=n;i++){ scanf("%d",&in[i]); pt[in[i]]=i; } for(int i=1;i<=n;i++){ scanf("%d",&pre[i]); } while(q--){ int a,b; scanf("%d%d",&a,&b); if(pt[a]==0&&pt[b]==0){ printf("ERROR: %d and %d are not found.n",a,b); } else if(pt[a]==0||pt[b]==0){ printf("ERROR: %d is not found.n",pt[a]==0?a:b); } else { ok=0; lca(1,n,1,a,b); } } return 0;}