求蔓延病毒最多可以感染的城市数的问题

原题链接
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即可判断是否两点不存在公共祖先(该题链接)。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值