codeforces E. Jamie and Tree LCA+dfs序+线段树

题解:

写起来还稍微有点麻烦。
dfs序+线段树可以维护子树的整体修改和查询。 因此,这道题我们要往子树上靠。
我们首先从1号点进行dfs遍历,顺便求出点的dfs序和深度,然后我们采用倍增的思想,可以预处理出每个点的祖先是谁。然后可以在O(log(n))O(log(n))的时间复杂度内求出任意两点的lca(u,v)lca(u,v)
而现在整个树的根是可以改变的,因此,我们需要一个结论,也就是说当树的根节点被改变为root时候,u和v的新的lca,也就是newlca(u,v)=lca(u,v)xorlca(u,root)xorlca(v,root)newlca(u,v)=lca(u,v)xorlca(u,root)xorlca(v,root) (这个可以自己画画图看一下)。
找到newlca以后还不行,根据newlca与root的关系不一样,还需要进一步讨论。
1. 当newlca=rootnewlca=root的时候,要操作的子树就是整颗树。
2. 当lca(newlca,root)!=newlcalca(newlca,root)!=newlca 那么要操作的子树就是以1为根节点时候的newlca的子树。
3. 当lca(newlca,root)==newlcalca(newlca,root)==newlca的时候,那么要操作的就是整颗树减去以(root到newlca链上深度为dep[newlca]-1的)点的子树。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define int long long
#define pr(x) cout<<#x<<":"<<x<<endl
const int maxn = 1e5+7;
int n,q;
struct edge{int u,v,nxt;
}es[maxn<<1];
int head[maxn];
int tot = 0;
void addedge(int u,int v){es[tot].u = u,es[tot].v = v,es[tot].nxt = head[u];head[u] = tot++;
}
int fa[maxn][22],dep[maxn];
int idx = 0,IN[maxn],OUT[maxn];
int segtree[maxn<<3],addmark[maxn<<3];
void init(){memset(fa,0,sizeof(fa));memset(head,-1,sizeof(head));memset(IN,0,sizeof(IN));memset(OUT,0,sizeof(OUT));memset(dep,0,sizeof(dep));tot = idx = 0;
}
void dfs(int u,int myfa,int dp){dep[u] = dp;IN[u] = ++idx;for(int e = head[u];e != -1;e = es[e].nxt){int v = es[e].v;if(v == myfa) continue;fa[v][0] = u;dfs(v,u,dp+1);}OUT[u] = ++idx;
}
void pushdown(int rt,int lft,int rgt){if(addmark[rt]){int mid = (lft + rgt)/2;addmark[2*rt] += addmark[rt];addmark[2*rt+1] += addmark[rt];segtree[2*rt] += addmark[rt]*(mid-lft+1);segtree[2*rt+1] += addmark[rt]*(rgt-mid);addmark[rt] = 0;}
}
void pushup(int rt){segtree[rt] = segtree[rt*2] + segtree[rt*2+1];
}
int val[maxn];
/*
void build(int rt,int L,int R){if(R == L) {segtree[rt] = val[L];}else{build(L,mid);build(mid+1,R);}
}*/
void ins(int rt,int lft,int rgt,int L,int R,int adv){if(rgt < L || lft > R) return ;if(L <= lft && R >= rgt) {segtree[rt] += (rgt-lft+1)*adv;addmark[rt] += adv;return ;}int mid = (lft + rgt) / 2;pushdown(rt,lft,rgt);ins(rt*2,lft,mid,L,R,adv);ins(rt*2+1,mid+1,rgt,L,R,adv);pushup(rt);
}
int ask(int rt,int lft,int rgt,int L,int R){if(rgt < L || lft > R) return 0;if(L <= lft && R >= rgt) return segtree[rt];pushdown(rt,lft,rgt);int mid = (lft + rgt)/2;return ask(rt*2,lft,mid,L,R) + ask(rt*2+1,mid+1,rgt,L,R);
}
void makelca(){for(int i = 1;i < 20;++i){for(int u = 1;u <= n;++u){fa[u][i] = fa[fa[u][i-1]][i-1];}}
}
int lca(int u,int v){if(dep[u] < dep[v]) swap(u,v);int dpc = dep[u] - dep[v];if(dpc){int t = 0;while(dpc){if(dpc & 1)u = fa[u][t];t++;dpc >>= 1;}}if(u == v) return u;for(int i = 19;u != v && i >= 0;--i ){if(fa[u][i] != fa[v][i]) {u = fa[u][i];v = fa[v][i];}}return fa[u][0];
}int root = 1;
main(){init();scanf("%lld%lld",&n,&q);for(int i = 1;i <= n;++i){scanf("%lld",&val[i]);}for(int i = 0;i < n-1;++i){int u,v;scanf("%lld%lld",&u,&v);addedge(u,v);addedge(v,u);}dfs(1,-1,0);makelca();for(int i = 1;i <= n;++i){ins(1,1,2*n,IN[i],IN[i],val[i]);ins(1,1,2*n,OUT[i],OUT[i],val[i]);}//计算lcafor(int i = 0;i < q;++i){int op ;scanf("%lld",&op);if(op == 1){scanf("%lld",&root);}else if(op == 2){int u,v,x;scanf("%lld%lld%lld",&u,&v,&x);int rt = lca(u,v)^lca(u,root)^lca(root,v);if(rt == root) ins(1,1,2*n,1,2*n,x);else if(lca(rt,root) != rt) ins(1,1,2*n,IN[rt],OUT[rt],x);else{int dpc = dep[root]-dep[rt]-1;int t = 0;int tmp = root;while(dpc){if(dpc&1)tmp = fa[tmp][t];t++;dpc >>= 1;}//cout<<tmp<<' '<<IN[tmp]<<' '<<OUT[tmp]<<endl;ins(1,1,2*n,1,2*n,x);ins(1,1,2*n,IN[tmp],OUT[tmp],-x);}}else if(op == 3){int v;scanf("%lld",&v);if(v == root){printf("%lld\n",ask(1,1,2*n,1,2*n)/2);}else if(lca(v,root) != v){printf("%lld\n",ask(1,1,2*n,IN[v],OUT[v])/2);}else{int tmp = root;int dpc = dep[root]-1-dep[v];int t = 0;while(dpc){if(dpc&1)tmp = fa[tmp][t];++t;dpc >>= 1;}int ans = (ask(1,1,2*n,1,2*n)-ask(1,1,2*n,IN[tmp],OUT[tmp]))/2;printf("%lld\n",ans);}}}//return 0;
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/321351.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

async/await 的基本实现和 .NET Core 2.1 中相关性能提升

前言这篇文章的开头&#xff0c;笔者想多说两句&#xff0c;不过也是为了以后再也不多嘴这样的话。在日常工作中&#xff0c;笔者接触得最多的开发工作仍然是在 .NET Core 平台上&#xff0c;当然因为团队领导的开放性和团队风格的多样性&#xff08;这和 CTO 以及主管的个人能…

【DP】砝码称重 (ssl 1072)

砝码称重 ssl 1072 题目大意&#xff1a; 有6种砝码&#xff0c;每种的个数分别为a1,a2,a3,a4,a5,a6,请问可以平出多少种重量(0除外) Description 设有1g、2g、3g、5g、10g、20g的砝码各若干枚&#xff08;其总重<1000&#xff09;&#xff0c; 要求&#xff1a; 输入…

P5025-[SNOI2017]炸弹【tarjan,线段树优化建图】

正题 题目链接:https://www.luogu.com.cn/problem/P5025 题目大意 .nnn个炸弹&#xff0c;每个在xxx位置处&#xff0c;范围为rrr。定义fif_ifi​表示第iii个炸弹爆炸能连锁的炸弹数&#xff0c;要求输出∑i1nfi∗i\sum_{i1}^nf_i*ii1∑n​fi​∗i 解题思路 将每个炸弹可以炸…

使用Swashbuckle构建RESTful风格文档

本次和大家分享的是Swagger to WebApi的nuget包Swashbuckle&#xff1b;因为项目需要统一api文档的风格&#xff0c;并要支持多种开发语言&#xff08;C#&#xff0c;java&#xff0c;python&#xff09;&#xff0c;所以首先想到的是swagger来构建api文档&#xff0c;本章讲解…

【dfs】【bfs】【链表】 求连通分量 (ssl 1759)

求连通分量 ssl 1759 题目大意 由n个点组成的无向图&#xff0c;求连通在一起的点数最大是多少 原题 求一个图的连通分量 Input n 顶点数(<100) 边 Output 连通分量 Sample Input 8 6 3 1 2 2 5 5 4 4 1 8 7 0 0 Sample Output 4 方法一&#xff08;dfs …

P2472-[SCOI2007]蜥蜴【网络流】

正题 题目链接:https://www.luogu.com.cn/problem/P2472 题目大意 n∗mn*mn∗m个格子&#xff0c;每个格子的石柱高度不同&#xff0c;蜥蜴可以跳到距离不超过ddd的石柱处&#xff0c;并且先前所站的石柱高度减一&#xff0c;为0则不能站&#xff0c;然后求有多少只蜥蜴不可以…

发布 Rafy .NET Standard 版本 Nuget 包

去年年中&#xff0c;Rafy 框架的源码就已经支持了 Net Standard 2.0 版本。其开源代码也已经上传到 Github 中&#xff1a;https://github.com/zgynhqf/rafy/tree/NetStandard2.0 。但是这都只是在源码层面支持 NS2.0&#xff0c;并没有发布其正式的 Nuget 包。要使用这个版本…

一些来自STL的好东西

STL: 队列: 表达式作用#include&#xff1c;queue&#xff1e;定义queue&#xff1c;int&#xff1e;x定义一个int类型的队列&#xff0c;名为xx.push(y)从队列x的对尾插入yx.pop()队列x的队头出列gx.front()g等于x的队头hx.back()h等于x的队尾x.size()队列x的长度x.empty()判…

codeforces D.MADMAX 动态规划、记忆化搜索

题意 给出一个DAG&#xff0c;每条边上有权重(权重是小写字母的ASCII码)&#xff0c;现在两位同学A和B分别位于某两点上(可以相同)&#xff0c;其中A和B轮流走&#xff0c;但是每人所走的边权不能变小&#xff0c;走到不能走为止就输。 A先走&#xff0c;询问最后谁会赢。 题解…

GDOI2020游记

DAY0DAY0DAY0 早上上课一堆作业没写完(啊这 然后中午去竞赛室之前还没找到老师请假&#xff0c;还专门又跑回去拿了准考证(还白嫖了马老师红包 然后下午就去了&#xff0c;发现没带充电线 之后发现准考证没带&#xff0c;吓死了。还好酒店有可以打印的地方&#xff0c;还好搞定…

你关心才值得分享 | K8S网络安全之访问控制技术实践

(请允许我插播下广告&#xff0c;便于其它伙伴了解趣码 Cloud Coder)还是那句话&#xff0c;你关心才值得分享~最近的一起分享就在5.10本周四晚&#xff0c;精彩千万不要错过&#xff01;Hi&#xff0c;你是不是也曾觉得K8S&#xff08; Kubernetes &#xff09;网络安全话题范…

【dfs】【链表】连通图 (ssl 1758)

连通图 ssl 1758 题目大意 有一个由n个点组成的无向图&#xff0c;检测他是否联通 原题 判断一个图是否为一个边通图 Input n 顶点 (n<100) 边 Output 1 表示连通 0 表示不边通 Sample Input 5 1 2 2 3 5 4 0 0 Sample Output 0 解题方法 用dfs链表从1开始…

洛谷P1120小木棒 爆搜+剪枝

题解 暴搜的思路容易想到&#xff0c;但是剪枝细节有很多&#xff0c;数据很强。 搜索思路&#xff1a; a. 用dfs(left_num,left_len,bound)表示当前还需要拼left_num根木棒&#xff0c;当前正在拼的木棒还剩left_len长度&#xff0c;搜索是从大往小搜索&#xff0c;并且当前搜…

P3338-[ZJOI2014]力【FFT】

正题 题目链接:https://www.luogu.com.cn/problem/P3338 题目大意 Fj∑i1j−1qi∗qj(i−j)2−∑ij1nqi∗qj(i−j)2F_j\sum_{i1}^{j-1}\frac{q_i*q_j}{(i-j)^2}-\sum_{ij1}^n\frac{q_i*q_j}{(i-j)^2}Fj​i1∑j−1​(i−j)2qi​∗qj​​−ij1∑n​(i−j)2qi​∗qj​​ EjFjqjE_j…

从Xamarin.Essentials谈Xamarin库的封装

编者语&#xff1a;Xamarin在国内的推广还需要努力&#xff0c;其实这真的是移动端开发的一大福音&#xff0c;毕竟用一份代码的时间可以生成iOS/Android/Windows/Linux/macOS/Tizen多个平台&#xff0c;而且是原生的性能。Xamarin在Build 2018发布的新功能有Xamarin.Essential…

【最短路】【图论】【Floyed】牛的旅行(ssl 1119/luogu 1522)

牛的旅行 ssl 1119 luogu 1522 题目大意 有两堆点&#xff0c;每一堆点之中的任何两个点都一定有相连的路线&#xff0c;连接两堆点中的各一个点&#xff0c;使最远的两个点的距离最短 原题 农民John的农场里有很多牧区。有的路径连接一些特定的牧区。一片所有连通的牧区称…

codefoces 939E Maximize!好题

题解 若存在一个子集s满足答案的话&#xff0c;则该子集一定包含集合S的最大值。 反证法证明&#xff1a; 假设s集合中最大的元素为x&#xff0c;S集合中最大的元素为X。则如果把x换成X&#xff0c;最大值增加了X-x&#xff0c;而平均值增量一定不大于X-x。 这样的话&#xff0…

P2050-[NOI2012]美食节【费用流,动态连边】

正题 题目链接:https://www.luogu.com.cn/problem/P2050 题目大意 nnn个菜品mmm个厨师&#xff0c;第iii种菜需要pip_ipi​份&#xff0c;第iii个人做第jjj道菜需要时间ti,jt_{i,j}ti,j​&#xff0c;求最少等待时间和。 解题思路 这题和之前修车很像&#xff0c;数据变大了。…

用ASP.NET Core 2.0 建立规范的 REST API -- 预备知识

什么是RESTREST 是 Representational State Transfer 的缩写. 它是一种架构的风格, 这种风格基于一套预定义的规则, 这些规则描述了网络资源是如何定义和寻址的.一个实现了REST这些规则的服务就叫做RESTful的服务.最早是由Roy Fielding提出的.RPC 风格/getUsers/getUser?id1/c…

【图论】【最短路】【Dijkstra】最小花费(ssl 2206/luogu 1576)

最小花费 ssl 2206 luogu 1576 题目大意&#xff1a; 有n个人&#xff0c;他们之间有m对人可以相互{\color{red}相互}相互转账&#xff0c;但要收一定的税&#xff0c;求第x个人转给第y个人至少要多少钱 Description 在n个人中&#xff0c;某些人的银行账号之间可以互相转…