[CF226E]Noble Knight's Path
[CF226E]Noble Knight's Path
题目大意：
一棵\(n(n\le10^5)\)个结点的树，初始时所有结点都是白色。\(m(m\le10^5)\)次操作，操作包含以下两种：
- 将点\(u\)涂黑。
- 询问从\(u\)到\(v\)的路径上，只考虑\(y\)以后的操作，第\(k\)个白色的结点（不包含\(u\)和\(v\)）。
思路：
树链剖分+主席树。
源代码：
#include<cstdio>
#include<cctype>
#include<algorithm>
#include<forward_list>
inline int getint() {
register char ch;
while(!isdigit(ch=getchar()));
register int x=ch^'0';
while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
return x;
}
constexpr int N=1e5+1,logN=17;
std::forward_list<int> e[N];
int dep[N],par[N],size[N],son[N],top[N],dfn[N],id[N];
void dfs1(const int &x) {
size[x]=1;
dep[x]=dep[par[x]]+1;
for(auto &y:e[x]) {
dfs1(y);
size[x]+=size[y];
if(size[y]>size[son[x]]) {
son[x]=y;
}
}
}
void dfs2(const int &x) {
id[dfn[x]=++dfn[0]]=x;
top[x]=x==son[par[x]]?top[par[x]]:x;
if(son[x]) dfs2(son[x]);
for(auto &y:e[x]) {
if(y!=son[x]) dfs2(y);
}
}
class FotileTree {
#define mid ((b+e)>>1)
private:
static constexpr int SIZE=N*logN*2;
struct Node {
int val,left,right;
};
Node node[SIZE];
int sz,new_node(const int &p) {
node[++sz]=node[p];
return sz;
}
int length(const int &b,const int &e) const {
return e-b+1;
}
public:
int root[N];
void insert(int &p,const int &b,const int &e,const int &x) {
p=new_node(p);
node[p].val++;
if(b==e) return;
if(x<=mid) insert(node[p].left,b,mid,x);
if(x>mid) insert(node[p].right,mid+1,e,x);
}
int query(const int &p,const int &q,const int &b,const int &e,const int &x) const {
if(node[p].val-node[q].val==0) return 1;
if(node[p].val-node[q].val==length(b,e)) return 0;
if(x<=mid) return query(node[p].left,node[q].left,b,mid,x);
return query(node[p].right,node[q].right,mid+1,e,x);
}
int query(const int &p,const int &q,const int &b,const int &e,const int &l,const int &r) const {
//printf("````%d %d %d %d %d\n",b,e,l,r,node[p].val-node[q].val);
if(node[p].val-node[q].val==0) return length(l,r);
if(node[p].val-node[q].val==length(b,e)) return 0;
if(b==l&&e==r) return length(b,e)-(node[p].val-node[q].val);
int ret=0;
if(l<=mid) ret+=query(node[p].left,node[q].left,b,mid,l,std::min(mid,r));
if(r>mid) ret+=query(node[p].right,node[q].right,mid+1,e,std::max(mid+1,l),r);
return ret;
}
int query(const int &p,const int &q,const int &b,const int &e,const int &l,const int &r,const int &k) const {
if(b==e) return id[b];
if(r<=mid) return query(node[p].left,node[q].left,b,mid,l,r,k);
if(l>mid) return query(node[p].right,node[q].right,mid+1,e,l,r,k);
const int tmp=query(node[p].left,node[q].left,b,mid,l,mid);
if(tmp>=k) return query(node[p].left,node[q].left,b,mid,l,mid,k);
return query(node[p].right,node[q].right,mid+1,e,mid+1,r,k-tmp);
}
#undef mid
};
FotileTree t;
int get_lca(int x,int y) {
while(top[x]!=top[y]) {
if(dep[top[x]]<dep[top[y]]) std::swap(x,y);
x=par[top[x]];
}
if(dep[x]<dep[y]) std::swap(x,y);
return y;
}
int query(int x,int y,const int &p,const int &q) {
int ret=0;
while(top[x]!=top[y]) {
if(dep[top[x]]<dep[top[y]]) std::swap(x,y);
ret+=t.query(t.root[p],t.root[q],1,dfn[0],dfn[top[x]],dfn[x]);
x=par[top[x]];
}
if(dep[x]<dep[y]) std::swap(x,y);
ret+=t.query(t.root[p],t.root[q],1,dfn[0],dfn[y],dfn[x]);
return ret;
}
int query(int x,int y,int k,const int &p,const int &q) {
const int z=get_lca(x,y);
const int tmp1=t.query(t.root[p],t.root[q],1,dfn[0],dfn[x]);
const int sum1=query(x,z,p,q);
k+=tmp1;
if(sum1>=k) {
while(top[x]!=top[z]) {
const int tmp=t.query(t.root[p],t.root[q],1,dfn[0],dfn[top[x]],dfn[x]);
if(tmp<k) {
k-=tmp;
x=par[top[x]];
continue;
}
return t.query(t.root[p],t.root[q],1,dfn[0],dfn[top[x]],dfn[x],tmp-k+1);
x=par[top[x]];
}
const int sum=t.query(t.root[p],t.root[q],1,dfn[0],dfn[z],dfn[x]);
return t.query(t.root[p],t.root[q],1,dfn[0],dfn[z],dfn[x],sum-k+1);
} else {
k=query(x,y,p,q)-k+1;
while(top[y]!=top[z]) {
const int tmp=t.query(t.root[p],t.root[q],1,dfn[0],dfn[top[y]],dfn[y]);
if(tmp<k) {
k-=tmp;
y=par[top[y]];
continue;
}
return t.query(t.root[p],t.root[q],1,dfn[0],dfn[top[y]],dfn[y],tmp-k+1);
y=par[top[y]];
}
const int sum=t.query(t.root[p],t.root[q],1,dfn[0],dfn[z],dfn[y]);
return t.query(t.root[p],t.root[q],1,dfn[0],dfn[z],dfn[y],sum-k+1);
}
}
int main() {
//freopen("travel.in","r",stdin);
//freopen("travel.out","w",stdout);
const int n=getint();
for(register int i=1;i<=n;i++) {
e[par[i]=getint()].emplace_front(i);
}
const int &root=*e[0].begin();
dfs1(root);
dfs2(root);
const int m=getint();
for(register int i=1;i<=m;i++) {
t.root[i]=t.root[i-1];
const int opt=getint();
if(opt==1) {
t.insert(t.root[i],1,n,dfn[getint()]);
}
if(opt==2) {
const int u=getint(),v=getint(),k=getint(),y=getint();
const int tmp=t.query(t.root[i],t.root[y],1,n,dfn[u])+t.query(t.root[i],t.root[y],1,n,dfn[v]);
//printf("````%d %d %d\n",query(u,v,i,y),tmp,k);
if(query(u,v,i,y)-tmp<k) {
puts("-1");
continue;
}
printf("%d\n",query(u,v,k,i,y));
}
}
return 0;
}
