G
树上DFS + set启发式合并
题意:
给定一棵树和一个排列p , 给定若干次询问 每次询问给出l,r,x
求是否p[l]~p[r]有一个结点的祖先是x
思路:
不妨做个映射,把每个结点的编号映射为它在排列中的下标
那么转化为求x的子树中是否有l~r结点存在
考虑怎么判断:
对于给定的l,r,x ,若知道x的子树结点的集合,那么只需二分判断是否存在>=l并且<=r的元素存在即可
我们做一个DFS , 每次找出每个结点的重儿子,先将它的set转移而非合并到此节点的set中,其他节点的set暴力合并即可
复杂度O(n(logn)^2)
void solve(){int n,q;cin>>n>>q;vector<pii>edge;rep(i,1,n-1){int u,v;cin>>u>>v;edge.pb({u,v});}vector<int>p(n+1);vector<vector<pair<pii,int>>>qr(n+1);rep(i,1,n){int x;cin>>x;p[x]=i;}rep(i,1,q){int l,r,x;cin>>l>>r>>x;qr[p[x]].pb({{l,r},i});}vector<set<int>>s(n+1);vector<vector<int>>e(n+1);vector<int>ans(q+1,0);for(auto[l,r]:edge){e[p[l]].pb(p[r]);e[p[r]].pb(p[l]);}auto dfs =[&](auto&&self,int u,int fa)->void{int h=u;for(int v:e[u]){if(v==fa)continue;self(self,v,u);if(s[v].size()>s[h].size())h=v;}if(h!=u)swap(s[u],s[h]);s[u].insert(u);for(int v:e[u]){if(v==fa||v==h)continue;s[u].insert(s[v].begin(),s[v].end());}for(auto[PII,idx]:qr[u]){auto[l,r]=PII;auto px = s[u].lower_bound(l);if(px!=s[u].end() && (*px)<=r)ans[idx]=1;}};dfs(dfs,p[1],p[1]);rep(i,1,q){if(ans[i]==1)cout<<"YES"<<endl;else cout<<"NO"<<endl;}
}