# 树上差分（LCA） —— 题目: [闇の連鎖（暗的连锁）][松鼠的新家]

——4月11日

首先是关于……emmm
还是看看参考博客吧
膜拜Orz
文章： 树上差分算法笔记
博主： 秦淮岸灯火阑珊

然后老老实实打题吧，时间不够 ……
时间啊，还是不太够呢……
差点明明已经 就卡在边和点这种奇怪的地方了呢~ - ^ -

一、闇の連鎖

题目：闇の連鎖
这道题是关于边上权值转移的树上差分。下面那道题是关于点的权值转移的差分。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10, M = 2 * N;
int n, m;
int dep[N], fa[N][17];
int d[N], q[N], ans;
int h[N], ne[M], e[M], idx;
void add(int a, int b) {
	e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}

void bfs()
{
    memset(dep, 0x3f, sizeof dep);
    dep[0] = 0, dep[1] = 1;
    int hh = 0, tt = 0;
    q[0] = 1;

    while (hh <= tt)
    {
        int t = q[hh ++ ];
        for (int i = h[t]; ~i; i = ne[i])
        {
            int j = e[i];
            if (dep[j] > dep[t] + 1)
            {
                dep[j] = dep[t] + 1;
                q[ ++ tt] = j;
                fa[j][0] = t;
                for (int k = 1; k <= 16; k ++ )
                    fa[j][k] = fa[fa[j][k - 1]][k - 1];
            }
        }
    }
}

int lca(int a, int b)
{
    if (dep[a] < dep[b]) swap(a, b);
    for (int k = 16; k >= 0; k -- )
        if (dep[fa[a][k]] >= dep[b])
            a = fa[a][k];
    if (a == b) return a;
    for (int k = 16; k >= 0; k -- )
        if (fa[a][k] != fa[b][k])
        {
            a = fa[a][k];
            b = fa[b][k];
        }
    return fa[a][0];
}
int dfs(int u, int p) {
	int res = d[u];
	for(int i = h[u]; ~i; i = ne[i]) {
		int j = e[i];
		if(j == p) continue;
		int s = dfs(j, u);
		//恭喜找到一条砍完后乱砍都可以斩断连锁的边 
		if(s == 0) ans += m;
		else if(s == 1) ans ++;
		res += s;
	}
	return res;
}

int main() {
	scanf("%d%d", &n, &m);
	memset(h, -1, sizeof h);
	int a, b;
	for(int i = 1; i < n; i ++) {
		scanf("%d%d", &a, &b);
		add(a, b), add(b, a);
	}
	
	bfs();
	
	for(int i = 1; i <= m; i ++){
		scanf("%d%d", &a, &b);
		int l = lca(a, b);
		//d[x]是指当前节点x与其父节点相连的边被覆盖的次数
		//之所以是边……因为砍得就是边啊…… 
		d[a] ++, d[b] ++, d[l] -= 2;
	}
	dfs(1, -1);
	cout << ans;
	return 0;
}

二、P3258 [JLOI2014]松鼠的新家

题目：P3258 [JLOI2014]松鼠的新家
关于点的权值转移的差分。

#include<bits/stdc++.h>
using namespace std;

const int N = 300010, M = N * 2;

int n, m, vis[N], ans[N];
int h[N], e[M], ne[M], idx;
int dep[N], fa[N][30];
int d[N];
int q[N];

void add(int a, int b)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}

//bfs和lca这两个函数调了一个晚上……QAQ 
void bfs(int root) {
	dep[root] = 0;
	queue<int> q;
	q.push(root);
	while(q.size()) {
		int t = q.front();
		q.pop();
		for(int i = h[t]; ~i; i = ne[i]) {
			int j = e[i];
			if(j == fa[t][0]) continue;
			dep[j] = dep[t] + 1;
			fa[j][0] = t;
			q.push(j);
			for(int k = 1; k <= (int)log2(dep[j]); k ++) 
				fa[j][k] = fa[fa[j][k - 1]][k - 1];
		}
	}
}

int lca(int x, int y) {
	if(dep[x] < dep[y]) swap(x, y);
	while(dep[x] > dep[y]) {
		int t = (int)log2(dep[x] - dep[y]);
		x = fa[x][t];
	}
	if(x == y) return x;
	for(int t = (int)log2(dep[x]); t >= 0; t --) {
		if(fa[x][t] != fa[y][t]){
			x = fa[x][t], y = fa[y][t];
		}
	}
	return fa[x][0];
}

int dfs(int u, int father)
{
    int res = d[u];
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (j == father) continue;
        int s = dfs(j, u);
        res += s;
    }
    return ans[u] = res;
}

int main() {
    memset(h, -1, sizeof h);
    scanf("%d", &n);
    for(int i = 1; i <= n; i++) cin >> vis[i];
    for (int i = 1; i < n; i ++ ) {
        int a, b;
        scanf("%d%d", &a, &b);
        add(a, b), add(b, a);
    }
	int rot = vis[1];
    bfs(rot);
	
    for (int i = 1; i < n; i ++ )
    {
    	int a = vis[i], b = vis[i + 1];
        int p = lca(a, b);
        d[a] ++, d[b] ++, d[p] -= 1, d[fa[p][0]] -= 1;
    }
    dfs(rot, -1);
    //将重复的都减掉(除了根节点，其它点出的时候不消耗糖果,就是说会有一次是不用糖果的，但是根节点每次进去都要消耗糖果…… 
    for(int i = 2; i <= n; i ++) ans[vis[i]] --;
    for(int i = 1;i <= n; i++) printf("%d\n", ans[i]);
    return 0;
}