题目描述

植物学家Dustar培养出了一棵神奇的树,这棵有根树有n个节点,每个节点上都有一个数字a[i],而且这棵树的根为r节点。
这棵树非常神奇,可以随意转换根的位置,上一秒钟它的根是x节点,下一秒钟它的根就变成了y节点,Dustar由于魔力不够,无法控制住这棵树,这棵树说:“只要你每次都能正确答出以x为根的子树里的点的权值和我就停下来。”,你能帮帮Dustar吗?

题目大意

给你一棵树根不定的树,每次让你查询\(x\)节点的子树权值和。

解法

树链剖分求换根树。
对于每一个\(u\),我们假设\(root\)是当前整棵树的根。

  • 如果\(u=root\),那么答案就是整个树的权值和
  • 如果\(Lca(u,root)≠u\),就说明u的子树没有变,答案还是在原来的u的答案。
  • 如果\(Lca(u,root)=u\),即\(u\)在原来的树中是root 的祖先。那么我们找到\(u\)到\(root\)路径上的第一个儿子。这个儿子对应的原树中的子树,就是现在\(u\)的子树的补集。

ac代码

#include<iostream>
#define N 200005
#define lson nod<<1
#define rson nod<<1|1
using namespace std;
struct edge{int to,nt;};
edge E[N<<1];
int H[N],v[N],sz[N],fa[N],top[N],dep[N],son[N],pos[N],tot,idx[N],out[N],tree[N<<2],val[N],add[N<<2];
int cnt,n,m,rt;
int r(){
    int w=0,x=0;char ch=0;
    while(!isdigit(ch))w|=ch=='-',ch=getchar();
    while(isdigit(ch))x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
    return w?-x:x;
}
void pushup(int nod){tree[nod]=tree[lson]+tree[rson];}
void pushdown(int l,int r,int nod){
    if(!add[nod]) return;
    int mid=l+r>>1;
    add[lson]+=add[nod];
    add[rson]+=add[nod];
    tree[lson]+=(mid-l+1)*add[nod],tree[rson]+=(r-mid)*add[nod];
    add[nod]=0;
}
void addedge(int u,int v){
    E[++cnt]=(edge){v,H[u]};H[u]=cnt;
    E[++cnt]=(edge){u,H[v]};H[v]=cnt;
}
void dfs1(int u){
    sz[u]=1;
    for(int i=H[u];i;i=E[i].nt){
        int v=E[i].to;
        if(v==fa[u]) continue;
        fa[v]=u,dep[v]=dep[u]+1;
        dfs1(v); sz[u]+=sz[v];
        if(sz[v]>sz[son[u]]) son[u]=v;
    }
}
int Lca(int u,int v){//求最近公共祖先
    while(top[u]!=top[v]) {
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        u=fa[top[u]];
    }
    return dep[u]>dep[v]?v:u;
}
void dfs2(int u,int tp) {//求重边,dfs序
    pos[u]=++tot,idx[tot]=u;top[u]=tp;
    if(!son[u]) return void(out[u]=tot);
    dfs2(son[u],tp);
    for(int i=H[u];i;i=E[i].nt){
        int v=E[i].to;
        if(v==fa[u]||v==son[u]) continue;
        dfs2(v,v);
    }
    out[u]=tot;
}
void build(int l,int r,int nod){//建线段树
    if(l==r){tree[nod]=val[idx[l]]; return;}
    int mid=l+r>>1;
    build(l,mid,lson); build(mid+1,r,rson);
    pushup(nod);
}
int query(int l,int r,int ql,int qr,int nod){
    if(ql<=l&&r<=qr)return tree[nod];
    pushdown(l,r,nod);
    int res=0,mid=l+r>>1;
    if(ql<=mid) res+=query(l,mid,ql,qr,lson);
    if(qr>mid) res+=query(mid+1,r,ql,qr,rson);
    pushup(nod);
    return res;
}
int find(int u,int v){//找到u和v上的第一个儿子
    while(top[u]!=top[v]) {
        if(dep[top[u]]<dep[top[v]]) swap(u,v);
        if(fa[top[u]]==v) return top[u];
        u=fa[top[u]];
    }
    if(dep[u]<dep[v]) swap(u,v);
    return son[v];
}
int subquery(int u){
    if(u==rt) return query(1,n,1,n,1);
    int lca=Lca(u,rt);
    if(lca!=u) return query(1,n,pos[u],out[u],1);
    int son=find(u,rt),res=0;
    res+=query(1,n,1,n,1);
    res-=query(1,n,pos[son],out[son],1);
    return res;
}
int main(){
    n=r(),m=r();
    for(int i=1;i<n;i++)addedge(r(),r());
    for(int i=1;i<=n;i++)val[i]=r();
    rt=r(); dfs1(rt); dfs2(rt,rt);
    build(1,n,1);
    while(m--){
        int opt=r();
        if(opt==1)rt=r();
        if(opt==2)printf("%d\n",subquery(r()));
    }
    return 0;
}

[hgoi#2019/2/16t4]transform的更多相关文章

  1. 「HGOI#2019.4.19省选模拟赛」赛后总结

    t1-Painting 这道题目比较简单,但是我比较弱就只是写了一个链表合并和区间DP. 别人的贪心吊打我的DP,嘤嘤嘤. #include <bits/stdc++.h> #define ...

  2. [hgoi#2019/3/21]NOIP&NOI赛后总结

    前言 今天做的是是2010年提高组和NOI的题目,做过几道原题,但是还是爆炸了,我真的太弱了. t1-乌龟棋 https://www.luogu.org/problemnew/show/P1541 这 ...

  3. [hgoi#2019/3/10]赛后总结

    关于本次hg模拟赛,题目来源于CF1110. t1-无意义运算符(meaning) 题目描述 最大公约数和位运算之间有共同点吗?是时候来研究一下了. 给定一个正整数a,请找到一个闭区间[1,a-1] ...

  4. [hgoi#2019/3/3]赛后总结

    T1--最长公共前缀(lcp) 定义两个字符串S,T 的最长公共前缀lcp(S,T)为最长的字符串R,满足R 既是S 的前缀又是T 的前缀. 给定一个字符串S,下标从1 开始,每次询问给出四个正整数a ...

  5. [hgoi#2019/2/16t3]psolve

    题目描述 Dustar有n道题目要做.他的月薪是m元. 由于题目是一流的难题,所以Dustar不得不找个人来帮(代)助(替)他写作业. 找人写作业不是免费的,但是他们能保证在一个月内做出任何题目.每做 ...

  6. [hgoi#2019/2/16t2]friend

    题目描述 在一个遥远的国度里有n个人,每个人手上写着4个互不相同的数. 这个国度比较奇怪,如果两个人至少有一个数字相同,则他们是一对朋友. 现在这n个人按序号从左到右排成了一排,每个人都想知道在他左边 ...

  7. [hgoi#2019/2/16t1]math

    题目描述 解法 我们稍微枚举一下前面几位,可以得到这样的规律. \[X_i=\frac{1}{2^{i+1}-1}\] \[Y_i=\frac{1}{2^{2^i}-1}\] 那么要使\(xm=yn\ ...

  8. [hgoi#2019/2/24]玄学考试

    感想 对于这次考试,真的不想说什么了,太玄学了!!! t1输出比标准输出长,这是什么操作???难道要关文件???但是交到oj上又A掉了.这是什么操作. t2还好,没有出什么意外...但是要吐槽一下出题 ...

  9. [hgoi#2019/2/18]比较水

    T1--调换纸牌(card) Alex有 n张纸牌,每张纸牌上都有一个值ai,Alex把这些纸牌排成一排,希望将纸牌按值从小到大的顺序排好.现在他把这个任务交给你,你只能进行一种操作:选中一张牌,然后 ...

随机推荐

  1. mac安装CocoaPods遇到的问题及解决办法

    (1)sudo gem install cocoapods Fetching: i18n-0.7.0.gem (100%) Successfully installed i18n-0.7.0 Fetc ...

  2. CSV文件解析

    CSV(逗号分隔值文件格式)        逗号分隔值(Comma-Separated Values,CSV,有时也称为字符分隔值,因为分隔字符也可以不是逗号),其文件以纯文本形式存储表格数据(数字和 ...

  3. Flask核心机制--上下文源码剖析

    一.前言 了解过flask的python开发者想必都知道flask中核心机制莫过于上下文管理,当然学习flask如果不了解其中的处理流程,可能在很多问题上不能得到解决,当然我在写本篇文章之前也看到了很 ...

  4. SRP周记_20190418

    SRP的全名是Student research project (我猜的),对于部分学生来说,不做就相当于不能毕业,一个project大概做一年左右,算是我工的一个特色.本人参加的SRP是做一个编译原 ...

  5. BTrace 初探

    BTrace 是一款java诊断工具,在解决现场问题的时候非常有用. 今天使用的时候碰到几个坑,先记录一下. 下载下来以后直接运行报错 root@iZ2ze89756yjbvq7le6obdZ:~/b ...

  6. Spring+SpringMVC+MyBatis整合(easyUI、AdminLte3)

    实战篇(付费教程) 花了几天的时间,做了一个网站小 Demo,最终效果也与此网站类似.以下是这次实战项目的 Demo 演示. 登录页: 富文本编辑页: 图片上传: 退出登录: SSM 搭建精美实用的管 ...

  7. ExtJS框架基础:事件模型及其常用功能

    前言 工作中用ExtJS有一段时间了,Ext丰富的UI组件大大的提高了开发B/S应用的效率.虽然近期工作中天天都用到ExtJS,但很少对ExtJS框架原理性的东西进行过深入学习,这两天花了些时间学习了 ...

  8. MVC_防止HttpPost重复提交

    重复提交的场景很常见,可能是当时服务器延迟的原因,如购物车物品叠加,重复提交多个订单.常见的解决方法是提交后把Button在客户端Js禁用,或是用Js禁止后退键等.在ASP.NET MVC 3 Web ...

  9. shell+curl监控网站页面(域名访问状态),并利用sendemail发送邮件

    应领导要求,对公司几个主要站点的域名访问情况进行监控.下面分享一个监控脚本,并利用sendemail进行邮件发送. 监控脚本如下:下面是写了一个多线程的网站状态检测脚本,直接从文件中读出站点地址,然后 ...

  10. C_数据结构_循环队列

    # include <stdio.h> # include <malloc.h> typedef struct Queue { int * pBase; int front; ...