洛谷P3258 [JLOI2014]松鼠的新家 树上差分

https://www.luogu.com.cn/problem/P3258

题目大意：给定一棵树，给n-1个起点与终点，路径上经过的点都需要加1。

树上差分_点差分模板题

点差分

 点差分时，俩点之间lca与其父-权值 俩点加权值。

 

边差分时，俩点加权值，lca-2*权值。

因为每个儿子都只有一个父亲所以边差分时对应的点为父亲->儿子。

最后权值计算应自底向上。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=300005;
#define int long long
vector<int>v[N];
vector<int>vc[N];
int f[N][31];
int a[N];
int depth[N];
int mark[N];
int n;
int ans[N];
void deal_first( int son , int father)//预处理 
{
    depth[son]=depth[father]+1;
    f[son][0]=father;
    for(int i=0;i<=19;i++)
    {
        f[son][i+1]=f[f[son][i]][i];
    }
    for(int i=0;i<v[son].size();i++)
    {
        if(v[son][i]==father)continue;
        deal_first(v[son][i],son);
    }
}
int lca(int x,int y)//x与y的公共最近祖先 
{
    if(depth[x]<depth[y])swap(x,y);
    for(int i=20;i>=0;i--)
    {
        if(depth[f[x][i]]>=depth[y])x=f[x][i];
        if(x==y)return x;
    }
    for(int i=20;i>=0;i--)
    {
        if(f[x][i]!=f[y][i]){
            x=f[x][i];
            y=f[y][i];
        }
    }
    return f[x][0];
}
int dfs(int son,int father){//自底向上计算 
	int num=mark[son];
	for(int i=0;i<v[son].size();i++){
		if(v[son][i]==father)continue;
		num+=dfs(v[son][i],son);
	}
	ans[son]+=num;
	return num;
}
signed main(){
	int last=0,now=0;
	scanf("%lld",&n);	
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
	}int fff=0;
	for(int i=1;i<n;i++){
		int x,y;
		scanf("%lld%lld",&x,&y);
		v[x].push_back(y);
		v[y].push_back(x); 
	}
	deal_first(1,0); 
	for(int i=1;i<n;i++){
		int la=lca(a[i],a[i+1]);
	//	printf("%d %d\n",f[la][0],la);
		mark[la]--;
		mark[f[la][0]]--;
		mark[a[i]]++;
		mark[a[i+1]]++;//差分标记的更新 
	}
	dfs(1,0);
	for(int i=2;i<=n;i++)ans[a[i]]--;
	for(int i=1;i<=n;i++){
		printf("%lld\n",ans[i]);
	}
}