[JLOI2014]松鼠的新家【树上差分】

Pro

QwQ

Sol

很裸的一道树上差分，不过和往常还不太一样。

根据题目中给出的ai来差分，最后肯定有点被多加了值，所以最后再跑一边去掉就好。

然而…… 我lca的模板打错了……

Code

#include
#include
#include
#include
using namespace std;

const long long L = 300005;
struct Node {
	long long to , next;
};
Node e[2*L];
long long n , q[L] , tot , head[L] , vis[L] , deep[L] , fa[L][35] , num[L];

void add(long long x , long long y) {
	tot++;
	e[tot].next = head[x];
	e[tot].to = y;
	head[x] = tot;
}

void dfs(long long u) {
	vis[u] = 1;
	for(int i=head[u]; i; i=e[i].next) {
		long long v = e[i].to;
		if(vis[v])
			continue;
		deep[v] = deep[u] + 1;
		fa[v][0] = u;
		dfs(v);
	}
}

long long lca(long long x , long long y) {
	if(deep[x]=0; i--)
		if(deep[fa[x][i]]>=deep[y])
			x = fa[x][i];
	if(x==y)
		return x;
	for(int i=30; i>=0; i--)
		if(fa[x][i]!=fa[y][i])
			x = fa[x][i] , y = fa[y][i];
	return fa[x][0];
}

void sol(long long l , long long r) {
	long long pa = lca(l,r);
	num[pa]--;
	num[fa[pa][0]]--;
	num[l]++;
	num[r]++;
}

void qans(long long u) {
	vis[u] = 1;
	for(int i=head[u]; i; i=e[i].next) {
		long long v = e[i].to;
		if(vis[v])
			continue;
		qans(v);
		num[u] += num[v];
	}
}

int main() {
	scanf("%lld",&n);
	for(int i=1; i<=n; i++)
		scanf("%lld",&q[i]);
	add(0 , 1);
	add(1 , 0);
	for(int i=1; i