bzoj 3631 [JLOI2014]松鼠的新家

题目

传送门

题解

看到某Yuta的树总结觉得LCA+树上差分挺好写而且差分没写过准备水一发结果水了一上午qwq。因为第一次写树上差分和树剖求LCA，结果树剖版子某处打错了。。（而且差分点和差分边还是有一定区别的）。

这题可以用树剖水部分分，不过n有30w，log方就会被卡了。所以利用差分思想把区间修改转化为点修改。给每个点一个tag表示新增值（貌似怪怪的）。如果有条路径（i，j）需要标记那么进行如下操作

tag（i）++；tag（j）++；tag（LCA（i，j））- -；tag（father（LCA（i，j）））- -；

这样能够达到标记经过的点的效果，请自行脑补如果是经过的边如何标记。
那么就把路径拆分就好了，但是这道题非起始端点会被多算一次（在衔接处），所以统计完后减回来。

题很水，人很蠢。

代码

//QWsin
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=600000+10;
const int maxm=1200000+10;
inline int read()
{int ret=0;char ch=getchar();while(ch<'0'||ch>'9') ch=getchar();for(;ch>='0'&&ch<='9';ch=getchar()) ret=ret*10+ch-'0';return ret;
}int to[maxn];int first[maxn],next[maxm],ecnt;
struct Edge{int u,v;Edge(int u=0,int v=0):u(u),v(v){}}e[maxm];
inline void add_edge(int u,int v)
{next[ecnt]=first[u];first[u]=ecnt;e[ecnt++]=Edge(u,v);next[ecnt]=first[v];first[v]=ecnt;e[ecnt++]=Edge(v,u);
}int top[maxn],hv[maxn],sz[maxn],p[maxn],dep[maxn];
int tag[maxn];  int dfs1(int u,int fa,int deep=1)
{p[u]=fa;dep[u]=deep;sz[u]=1;int msz=0;for(int i=first[u];i!=-1;i=next[i]) if(e[i].v!=fa){int lch=dfs1(e[i].v,u,deep+1);if(lch > msz) msz=lch,hv[u]=e[i].v;sz[u]+=lch;}return sz[u];
}void dfs2(int u,int Top)
{top[u]=Top;if(sz[u]==1) return ;dfs2(hv[u],Top);for(int i=first[u];i!=-1;i=next[i])if(e[i].v!=p[u]&&e[i].v!=hv[u]) dfs2(e[i].v,e[i].v);
}int LCA(int a,int b)
{for(;top[a]!=top[b];b=p[top[b]])if(dep[top[a]] > dep[top[b]]) swap(a,b);return dep[a]<dep[b]?a:b;
}int ans[maxn],sum,n;
int dfs3(int cur)
{int sum=tag[cur];for(int i=first[cur];i!=-1;i=next[i]) if(e[i].v!=p[cur]) sum+=dfs3(e[i].v);if(to[1]!=cur) ans[cur]=sum-1;else ans[cur]=sum; return sum;
}int main()
{n=read();memset(first,-1,sizeof(first));for(int i=1;i<=n;i++) to[i]=read();for(int i=1;i<n;i++) add_edge(read(),read());int root=1;dfs1(root,root);dfs2(root,root);tag[to[1]]++;tag[to[2]]++;int lca=LCA(to[1],to[2]);tag[lca]--;if(lca!=root) tag[p[lca]]--;for(int i=2;i<n;i++){int lca=LCA(to[i],to[i+1]);tag[to[i]]++;tag[to[i+1]]++;tag[lca]--;if(lca!=root) tag[p[lca]]--;}dfs3(root);for(int i=1;i<=n;i++) printf("%d\n",ans[i]);return 0;
}

bzoj 3631 [JLOI2014]松鼠的新家相关推荐

bzoj 3631: [JLOI2014]松鼠的新家（LCA+树上差分）
3631: [JLOI2014]松鼠的新家 Time Limit: 10 Sec Memory Limit: 128 MB Submit: 2059 Solved: 1030 [Submit][S ...
3631: [JLOI2014]松鼠的新家
3631: [JLOI2014]松鼠的新家 Time Limit: 10 Sec Memory Limit: 128 MB Submit: 707 Solved: 342 [Submit][Sta ...
3631. [JLOI2014]松鼠的新家【树形DP】
Description 松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有n-1根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的.天哪,他居然真的住在"树&q ...
【BZOJ 3631】 [JLOI2014]松鼠的新家
3631: [JLOI2014]松鼠的新家 Time Limit: 10 Sec Memory Limit: 128 MB Submit: 681 Solved: 329 [Submit][Statu ...
[Luogu 3258] JLOI2014 松鼠的新家
[Luogu 3258] JLOI2014 松鼠的新家 LCA + 树上差分. 我呢,因为是树剖求的 LCA,预处理了 DFN(DFS 序),于是简化成了序列差分. qwq不讲了不讲了,贴代码. #i ...
P3258 [JLOI2014]松鼠的新家
文章目录题意: 题解: 树上差分代码: 树链剖分代码: P3258 [JLOI2014]松鼠的新家题意: n个点,n-1条边,给出每个点的拜访顺序,问每个点经过几次(最后一次移动不算拜访) 题 ...
洛谷 P3258 [JLOI2014]松鼠的新家解题报告
P3258 [JLOI2014]松鼠的新家题目描述松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有n-1根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的.天哪,他 ...
洛谷 P3258 [JLOI2014]松鼠的新家树上差分
缘起 [1]中我们学习了树上差分,并且a了一个裸的点差分. 现在继续树上差分~ 洛谷 P3258 [JLOI2014]松鼠的新家分析题目描述松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房 ...
P3258[JLOI2014]松鼠的新家(LCA 树上差分)
P3258 [JLOI2014]松鼠的新家题目描述松鼠的新家是一棵树,前几天刚刚装修了新家,新家有n个房间,并且有n-1根树枝连接,每个房间都可以相互到达,且俩个房间之间的路线都是唯一的.天哪,他 ...

bzoj 3631 [JLOI2014]松鼠的新家

题目

题解

代码

bzoj 3631 [JLOI2014]松鼠的新家相关推荐

最新文章

热门文章