树上差分半裸题
常规思路是进行三次DFS,然后常规运算即可
这里提供两次dfs的思路(wyz tql orz)
我们以样例2为例
我们考虑任意一条路径,令其起点为u终点为v,每走一次当前路径则v的访问次数必定+1,于是我们可以使每一个点表示连接其上的一条边的访问次数,所以我们令节点v的访问次数+1;
与此同时,过程中的路径也同样会被访问,且这里是双向边,于是与此同时的我们也令节点u的访问次数+1;当然访问当前子树下根节点中包含的两个点并不会访问,而我们在增加u和v的访问时同时也错误地增加了其公共父节点的访问量,于是我们令lca(u,v)的访问量-2即可。
例如上图中我们从节点5走到节点3,我们令节点3与节点5的访问次数+1,同时使节点4的访问次数-2。
如下:
while(k--){
int u=read(),v=read();
diff[u]++,diff[v]++,diff[lca(u,v)]-=2;
}
最后输出答案时只需要判断每条边两端点的深度大小即可。
#include
#define int long long
#define maxn 100005
using namespace std;
inline char get(){
static char buf[30000],*p1=buf,*p2=buf;
return p1==p2 && (p2=(p1=buf)+fread(buf,1,30000,stdin),p1==p2)?EOF:*p1++;
}
inline int read(){
register char c=get();register int f=1,_=0;
while(c>'9' || c<'0')f=(c=='-')?-1:1,c=get();
while(c<='9' && c>='0')_=(_<<3)+(_<<1)+(c^48),c=get();
return _*f;
}
struct edge{
int u,v,w,next;
int num=0;
}E[maxn<<1];
int n,k;
int p[maxn],eid;
int d[maxn], parent[maxn][20];
int diff[maxn];
inline void init(){
for(register int i=0;i=0;j--){
if (d[x]-(1<=d[y])x=parent[x][j];
}
if(x==y)return x;
for(register int j=i;j>=0;j--){
if(parent[x][j]!=parent[y][j]) {
x=parent[x][j];
y=parent[y][j];
}
}
return parent[x][0];
}
int dd[maxn];
void dfs_(int u,int fa,int flag){
dd[u]=flag;
for(register int i=p[u];~i;i=E[i].next){
int v=E[i].v;
if(fa==v)continue;
dfs_(v,u,flag+1);
diff[u]+=diff[v];
}
}
int u[maxn],v[maxn];
signed main(){
//freopen("1.txt","r",stdin);
init();
n=read();
for(register int i=2;i<=n;i++){
u[i]=read(),v[i]=read();
insert2(u[i],v[i]);
}
d[1]=0;
dfs(1);
for(register int level=1;(1<=dd[v[i]])cout<