原题链接
n(<=1e5)个城市与n-1条边与q(<=1e5)个询问,每个城市有各自温度,距城市1越远温度越低,每次询问从x城市开始,蔓延一个生存温度在[l,r]区间的病毒,问最多感染多少个城市。
分析:
以城市建树。对于每个询问,从城市x开始倍增向上到温度小于等于r的最高城市nowX,答案就是:nowX的子节点数目-nowX子节点中温度小于l的数目。
求nowX子节点中温度小于l的数目,可以转换成求在nowX子节点中l的排名-1。
所以用树套树(或主席树或划分树)维护原树即可求得[dfn[nowX],dfn[nowX]+size[nowX]-1]中l的排名。
难点主要在用“树形结构”(树套树)维护“树形结构”(原树)的思路,要注意将原树的序列下标转换成对应的dfn值,然后再插入树套树中——这是和“树套树维护序列值”的用法区别。
struct Query{
int x,l,r;
bool operator <(const Query &a){return l<a.l;}
}query[maxv];
int tt,ans[maxv];
int dfn[maxv],tag=0,d[maxv],f[maxv][21],tem[maxv],siz[maxv];
vector<int> g[maxv];
bool vis[maxv];
void solveDfn(int p){
vis[p]=1,dfn[p]=++tag,siz[p]=1;
for(int i=0,y;i<g[p].size();i++)
if(!vis[y=g[p][i]]) solveDfn(y),siz[p]+=siz[y];
}
void bfs(){
queue<int> q;
q.push(1),d[1]=1;
while(q.size()){
int x=q.front();q.pop();
for(int i=0,y;i<g[x].size();i++){
if(d[y=g[x][i]]) continue;
d[y]=d[x]+1,f[y][0]=x;
for(int j=1;j<=tt;j++) f[y][j]=f[f[y][j-1]][j-1];
q.push(y);
}
}
}
struct Splay{/*此处省略Splay模板*/};
struct Segment_Tree
{
/*此处省略Segment_Tree模板*/
inline void LOL()
{
int n,x,l,r,u,v,q,now=1;
scanf("%d",&n),tt=(int)(log(n)/log(2))+1;
for(int i=1;i<=n-1;i++) scanf("%d %d",&u,&v),g[u].push_back(v),g[v].push_back(u);
solveDfn(1),bfs();
for(int i=1;i<=n;i++) scanf("%d",&tem[dfn[i]]);//注意要用dfn序!
build(root,1,n);
scanf("%d",&q);
for(int i=1;i<=q;i++) scanf("%d %d %d",&query[i].x,&query[i].l,&query[i].r);
for(int i=1;i<=q;i++){
int nowX=query[i].x; //注意要用dfn序!
if(tem[dfn[nowX]]<query[i].l||tem[dfn[nowX]]>query[i].r){ans[i]=0;continue;}
for(int j=20;j>=0;j--)
if(f[nowX][j]&&tem[dfn[f[nowX][j]]]<=query[i].r) nowX=f[nowX][j];
ans[i]=siz[nowX]-(rank(root,dfn[nowX],dfn[nowX]+siz[nowX]-1,query[i].l)+1-1);
}
for(int i=1;i<=q;i++) printf("%d\n",ans[i]);
}
}haha;
int main(){
haha.LOL();
return 0;
}
事实上我们经常用树形结构结合dfn值维护树形结构,如在判断一棵树某些的节点下标对应的另外一棵树节点下标是否有两点的一点是该两点的公共祖先上,可以用线段树维护节点在另外一棵树的dfn出序与入序,当加入一个节点,可add(入序,出序,1),这样判断总根的最大值dat是否为1即可判断是否两点不存在公共祖先(该题链接)。