hihoCoder1381 - Little Y's Tree
Description
给出一个\(n(n\leq10^5)\)个点的带边权的树。进行\(Q\)次询问:每次删除树上的\(k\)条边,求剩下的\(k+1\)个连通块中最远点对距离的和。\(\Sigma k\leq10^5\),询问之间是独立的。
Solution
神奇而又毒瘤的做法。
考虑如何合并树上两个连通块的答案。设两个连通块的最远点对分别为\((v_1,v_2),(v_3,v_4)\),那么合并后的最远点对的两个端点一定是\(\{v_1,v_2,v_3,v_4\}\)中的两个。LCA用RMQ求的话时间复杂度是\(O(1)\)。
证明:
设两个连通块通过边\((p,q)\)连通。若合并后的最远路径不经过\((p,q)\),则其一定是\((v_1,v_2),(v_3,v_4)\)之一。若经过\((p,q)\),则可以将其看成\((u_1,p)+(p,q)+(q,u_2)\),而\((v_1,p),(v_2,p)\)必然是以\(p\)为端点的最长、次长路径,所以\(u_1\)必然是\(v_1,v_2\)之一;\(u_2\)同理。
删除边\((u,v)\)相当于将以\(v\)为根的子树从原树上断掉,断掉\(k\)条边相当于将原树变成了以\(1\)和\(v_{1..k}\)为根的\(k+1\)个连通块。那么做出原树的DFS序,断掉一个子树就相当于删掉一个区间。如图,子树\(1\)中的子树\(2\)和子树\(6\)被断掉,那么就删掉这两个区间,剩下的\(\{1,3\}\)即以\(1\)为根的连通块;同理\(2\)中的\(4\)被断掉,从\(2\)的DFS序中删掉\(4\)的就是\(\{2,5\}\)。
对所有区间排序并递归,可以求出每个连通块中有哪些点,那么该连通块中的最远点对相当于DFS序上的若干个区间的合并。由于新加入一个区间最多把原区间分成三份,所以最多要询问\(2k+1\)次。用线段树维护DFS序,每个节点记录该区间内的最远点对即可。虽然DFS上连续的点在原树上不一定连通,不过由于我们每次询问的部分都是连通的所以没关系啦。
时间复杂度\(O(logn\Sigma k)\)。
Code
//Little Y's Tree
#include <algorithm>
#include <cstdio>
using std::sort; using std::max; using std::swap;
typedef long long lint;
inline char gc()
{
static char now[1<<16],*s,*t;
if(s==t) {t=(s=now)+fread(now,1,1<<16,stdin); if(s==t) return EOF;}
return *s++;
}
inline int read()
{
int x=0; char ch=gc();
while(ch<'0'||'9'<ch) ch=gc();
while('0'<=ch&&ch<='9') x=x*10+ch-'0',ch=gc();
return x;
}
const int N=1e5+10;
int n;
int cnt,h[N];
struct edge{int u,v,w,nxt;} ed[N<<1];
void edAdd(int u,int v,int w)
{
cnt++; ed[cnt].u=u,ed[cnt].v=v,ed[cnt].w=w,ed[cnt].nxt=h[u],h[u]=cnt;
cnt++; ed[cnt].u=v,ed[cnt].v=u,ed[cnt].w=w,ed[cnt].nxt=h[v],h[v]=cnt;
}
int fa[N],dpt[N]; lint dst[N];
int dfCnt1,dfn1[N],fr1[N],to1[N];
int dfCnt2,dfn2[N<<1],fr2[N];
void dfs(int u)
{
dfn1[++dfCnt1]=u; fr1[u]=dfCnt1;
dfn2[++dfCnt2]=u; fr2[u]=dfCnt2;
for(int i=h[u];i;i=ed[i].nxt)
{
int v=ed[i].v,w=ed[i].w;
if(v==fa[u]) continue;
fa[v]=u,dpt[v]=dpt[u]+1,dst[v]=dst[u]+w;
dfs(v); dfn2[++dfCnt2]=u;
}
to1[u]=dfCnt1;
}
int Lg2[N<<1],rmq[N<<1][20];
void bldLCA()
{
Lg2[1]=0;
for(int i=2;i<=dfCnt2;i++) Lg2[i]=Lg2[i>>1]+1;
for(int i=1;i<=dfCnt2;i++) rmq[i][0]=dfn2[i];
for(int k=1;k<=18;k++)
for(int i=1;i+(1<<k-1)<=dfCnt2;i++)
{
int r1=rmq[i][k-1],r2=rmq[i+(1<<k-1)][k-1];
rmq[i][k]=dpt[r1]<dpt[r2]?r1:r2;
}
}
int lca(int u,int v)
{
int i=fr2[u],j=fr2[v];
if(i>j) swap(i,j);
int t=Lg2[j-i+1];
int r1=rmq[i][t],r2=rmq[j-(1<<t)+1][t];
return dpt[r1]<dpt[r2]?r1:r2;
}
lint dist(int u,int v) {return dst[u]+dst[v]-2*dst[lca(u,v)];}
#define Ls (p<<1)
#define Rs (p<<1|1)
int rt=1; int maxP=0;
struct node
{
lint len; int v1,v2;
node(lint _len=0,int _v1=0,int _v2=0) {len=_len,v1=_v1,v2=_v2;}
}nd[N<<2];
node operator +(node x,node y)
{
if(x.v1==0) return y; else if(y.v1==0) return x;
node z=node(0,0,0);
lint d[10],d0=0;
d[1]=dist(x.v1,x.v2),d[2]=dist(x.v1,y.v1),d[3]=dist(x.v1,y.v2);
d[4]=dist(x.v2,y.v1),d[5]=dist(x.v2,y.v2),d[6]=dist(y.v1,y.v2);
for(int i=1;i<=6;i++) d0=max(d0,d[i]);
if(d[1]==d0) z=node(d[1],x.v1,x.v2);
else if(d[2]==d0) z=node(d[2],x.v1,y.v1);
else if(d[3]==d0) z=node(d[3],x.v1,y.v2);
else if(d[4]==d0) z=node(d[4],x.v2,y.v1);
else if(d[5]==d0) z=node(d[5],x.v2,y.v2);
else if(d[6]==d0) z=node(d[6],y.v1,y.v2);
return z;
}
void update(int p) {nd[p]=nd[Ls]+nd[Rs];}
void bldTr(int p,int L0,int R0)
{
maxP=max(maxP,p);
if(L0==R0) {nd[p]=node(0,dfn1[L0],dfn1[L0]); return;}
int mid=L0+R0>>1;
bldTr(Ls,L0,mid),bldTr(Rs,mid+1,R0);
update(p);
}
int optL,optR;
node query(int p,int L0,int R0)
{
if(optL<=L0&&R0<=optR) return nd[p];
int mid=L0+R0>>1; node res=node(0,0,0);
if(optL<=mid) res=res+query(Ls,L0,mid);
if(mid<optR) res=res+query(Rs,mid+1,R0);
return res;
}
struct qRec{int fr,to; node ans;} q[N];
bool cmpQ(qRec x,qRec y) {return x.fr<y.fr;}
int m,now;
//solve(x)解决区间x及其内部区间,并将now移动到x外的第一个
void solve(int x)
{
if(x>m) return;
now++; int pre=q[x].fr;
while(now<=m&&q[now].to<=q[x].to)
{
optL=pre,optR=q[now].fr-1;
if(optL<=optR) q[x].ans=q[x].ans+query(rt,1,n);
pre=q[now].to+1;
solve(now);
}
optL=pre,optR=q[x].to;
if(optL<=optR) q[x].ans=q[x].ans+query(rt,1,n);
}
int main()
{
n=read();
for(int i=1;i<=n-1;i++)
{
int u=read(),v=read(),w=read();
edAdd(u,v,w);
}
fa[1]=0,dfs(1);
bldLCA(); bldTr(rt,1,n);
int Q=read();
while(Q--)
{
m=read();
for(int i=1;i<=m;i++)
{
int x=read()<<1; int u=ed[x].u,v=ed[x].v;
if(dpt[u]>dpt[v]) swap(u,v);
q[i].fr=fr1[v],q[i].to=to1[v];
q[i].ans=node(0,0,0);
}
m++; q[m].fr=1,q[m].to=n,q[m].ans=node(0,0,0);
sort(q+1,q+m+1,cmpQ);
solve(now=1);
lint res=0;
for(int i=1;i<=m;i++) res+=(q[i].ans).len;
printf("%lld\n",res);
}
return 0;
}
P.S.
Icefox不到100行orz,我写了150+
hihoCoder1381 - Little Y's Tree的更多相关文章
- DFS序+线段树 hihoCoder 1381 Little Y's Tree(树的连通块的直径和)
题目链接 #1381 : Little Y's Tree 时间限制:24000ms 单点时限:4000ms 内存限制:512MB 描述 小Y有一棵n个节点的树,每条边都有正的边权. 小J有q个询问,每 ...
- [hihoCoder#1381]Little Y's Tree
[hihoCoder#1381]Little Y's Tree 试题描述 小Y有一棵n个节点的树,每条边都有正的边权. 小J有q个询问,每次小J会删掉这个树中的k条边,这棵树被分成k+1个连通块.小J ...
- Size Balance Tree(SBT模板整理)
/* * tree[x].left 表示以 x 为节点的左儿子 * tree[x].right 表示以 x 为节点的右儿子 * tree[x].size 表示以 x 为根的节点的个数(大小) */ s ...
- HDU3333 Turing Tree(线段树)
题目 Source http://acm.hdu.edu.cn/showproblem.php?pid=3333 Description After inventing Turing Tree, 3x ...
- Codeforces 620E New Year Tree(DFS序 + 线段树)
题目大概说给一棵树,树上结点都有颜色(1到60),进行下面两个操作:把某结点为根的子树染成某一颜色.询问某结点为根的子树有多少种颜色. 子树,显然DFS序,把子树结点映射到连续的区间.而注意到颜色60 ...
- Linux/Ubuntu tree 命令以树形结构显示文件夹目录结构
1.安装命令工具 sudo apt-get -y install tree 2.可以查看关于tree命令的帮助信息 $ tree --help usage: tree [-adfghilnpqrstu ...
- POJ3321 Apple Tree(DFS序)
题目,是对一颗树,单点修改.子树查询.典型的dfs序入门题. DFS序可以将一颗树与子树们表示为一个连续的区间,然后用线段树来维护:感觉算是树链剖分的一种吧,和轻重链剖分不同的是这是对子树进行剖分的. ...
- Bzoj 2588: Spoj 10628. Count on a tree 主席树,离散化,可持久,倍增LCA
题目:http://www.lydsy.com/JudgeOnline/problem.php?id=2588 2588: Spoj 10628. Count on a tree Time Limit ...
- poj3237 Tree
Description You are given a tree with N nodes. The tree’s nodes are numbered 1 through N and its edg ...
随机推荐
- 中国区 Azure 和全球版 Azure:功能对比
由世纪互联运营的 Microsoft Azure(文中简称为中国区 Azure)是在中国大陆独立运营的公有云平台,与全球其他地区由微软运营的 Azure (文中简称全球版 Azure)服务在物理上和逻 ...
- 面试题--JAVA中静态块、静态变量加载顺序
最后给大家一道面试题练练手,要求写出其结果(笔试) public class StaticTest { public static int k = 0; public static StaticTes ...
- Window10 开启传统启动界面
Windows 10沿袭了Windows 8的快速启动,导致在启动过程中无法通过按F8进入启动选项,这样当系统遇到问题无法进入时根本无法通过进入安全模式等方式进行处理(当然通过其他一些工具还是能够引导 ...
- Gym 100883J palprime(二分判断点在凸包里)
题意:判断一堆小点有多少个在任意三个大点构成的三角形里面. 思路:其实就是判断点在不在凸包里面,判断的话可以使用二分来判断,就是判断该点在凸包的哪两个点和起点的连线之间. 代码: /** @xigua ...
- 根据HTML语义化编码
语义化标签——http://www.html5jscss.com/html5-semantics-section.html 写HTML代码时应注意什么? 尽可能少的使用无语义的标签div和span: ...
- 转 在Qt中用QAxObject来操作Excel
最近写程序中需要将数据输出保存到Excel文件中.翻看<C++ GUI Programming with Qt 4>(Second Edition)发现可以在Qt中运用ActiveX控件, ...
- CPP-练习
HW: 1.局部变量能否和全局变量重名? 答:能,局部会屏蔽全局.要用全局变量,需要使用"::" ;局部变量可以与全局变量同名,在函数内引用这个变量时,会用到同名的局部变量,而不会 ...
- JS原型链(二)--new运算符的原理
new运算符的原理: 第一步:创建一个空对象,该对象继承构造函数的原型对象 第二步:执行这个构造函数,并且把this指向该空对象 第三步:返回:如果构造函数执行后返回的结果是一个object类型,则返 ...
- GIMP暗黑诱惑,部分彩色效果制作
在一些图形处理中经常会用到高逼格的部分彩色,其他部分黑白的效果,今天我就简单记录一下如何操作. 1.选区,先选择要突出的选区,可以用多种方法,钢笔,套绳,小剪刀等等: 2.把选择的区域稍稍调整亮一点: ...
- django第七天(模板的复用性,include标签和母版)
django第7天模板 include标签 模板的共用 a 模板需要到使用到 登陆界面 b 模板需要使用到 登陆界面 可以把登陆界面提取到公共的模板c 为什么要用: 都需要使用相同的界面,减少代码冗余 ...