bzoj3631（树链剖分或树形dp）

树链剖分，没有把线段树数组开大4倍！，卡了我好长时间！！！！

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=800009;
struct aa
{
    int to,pre;
}edge[maxn];
struct bb
{
    int add,l,r;
}a[maxn*2];
int size[maxn]={0},head[maxn]={0},tot=0,lab=0;
int tip[maxn]={0},top[maxn]={0},fa[maxn]={0},son[maxn]={0};
int n,order[maxn],dep[maxn]={0};
void addedge(int a,int b)
{
    edge[++tot].to=b;edge[tot].pre=head[a];head[a]=tot;
}
void dfs1(int u,int depth,int fat)
{
    fa[u]=fat;dep[u]=depth;
    int maxs=0;size[u]=1;
    for (int i=head[u];i;i=edge[i].pre)
    if (edge[i].to!=fa[u])
    {
        int v=edge[i].to;
        dfs1(v,depth+1,u);
        size[u]+=size[v];
        if (size[v]>maxs) maxs=size[v],son[u]=v;
    }
}
void dfs2(int u,int anc)
{
    top[u]=anc;tip[u]=++lab;
    if (son[u]!=0) dfs2(son[u],anc);
    for (int i=head[u];i;i=edge[i].pre)
    if (edge[i].to!=fa[u]&&edge[i].to!=son[u]) dfs2(edge[i].to,edge[i].to);
}
void buildtree(int i,int l,int r)
{
    a[i].l=l;a[i].r=r;
    if (l==r) return;
    int mid=(l+r)>>1;
    buildtree(i<<1,l,mid);
    buildtree(i<<1|1,mid+1,r);
}
void ins(int i,int l,int r)
{
    if (a[i].l==l&&a[i].r==r) {a[i].add++;return;}
    a[i<<1].add+=a[i].add;
    a[i<<1|1].add+=a[i].add;
    a[i].add=0;
    int mid=(a[i].l+a[i].r)>>1;
    if (mid>=r) ins(i<<1,l,r);
    else if (mid<l) ins(i<<1|1,l,r);
    else ins(i<<1,l,mid),ins(i<<1|1,mid+1,r);
}
void updata(int x,int y)
{
    while (top[x]!=top[y])
    {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        ins(1,tip[top[x]],tip[x]);
        x=fa[top[x]];
    }
    if (dep[x]>dep[y]) swap(x,y);
    ins(1,tip[x],tip[y]);
}
int find(int i,int x)
{
    if (a[i].l==a[i].r) return a[i].add;
    a[i<<1].add+=a[i].add;
    a[i<<1|1].add+=a[i].add;
    a[i].add=0;
    int mid=(a[i].l+a[i].r)>>1;
    if (mid>=x) find(i<<1,x);else find(i<<1|1,x);
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&order[i]);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addedge(x,y);addedge(y,x);
    }
    memset(a,0,sizeof(a));
    dfs1(1,1,0);
    dfs2(1,1);
    buildtree(1,1,n);
    for (int i=1;i<n;i++) updata(order[i],order[i+1]);
    int ans;
    for (int i=1;i<=n;i++) ans=find(1,tip[i]),
    printf("%d\n",i==order[1]? ans:ans-1);//重复的要减去
    return 0;
}

首先我们联想一下对数列进行差分的做法。序列差分这个技巧一般适用于：执行若干次区间加减，到最后再统计每个点的权值。设T是A的差分序列，我们把将a~b这个区间中的每个点加上c的操作（A[i]+=c  a<=i<=b）转变成对T的操作T[a]+=c，T[b+1]-=c，最后A[k]的值就是Sum{T[i]}(1<=i<=k)即T的一个前缀和。这个很容易能理解

然后树上差分，

#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstdio>
using namespace std;
const int maxn=800009;
int dep[maxn],f[maxn]={0},fa[maxn][22]={0},n,a[maxn],to[maxn],next[maxn],head[maxn]={0},tot=0,bj[maxn]={0};
void add(int a,int b)
{
	to[++tot]=b;next[tot]=head[a];head[a]=tot;
}
void dfs(int u,int depth)
{
	dep[u]=depth;
	int i=1;
	while ((1<<i)<depth)
	{
		fa[u][i]=fa[fa[u][i-1]][i-1];
		i++;
	}
	for (int i=head[u];i;i=next[i])
	if (to[i]!=fa[u][0]) fa[to[i]][0]=u,dfs(to[i],depth+1);
}
void up(int &x,int h)
{
	int i=0;
	while (h) 
	{
		if (h&1) x=fa[x][i];
		h>>=1;i++;
	}
}
void find(int x,int y)
{
	if (dep[x]<dep[y])swap(x,y);
	int xx=x,yy=y;
	up(x,dep[x]-dep[y]);
	
	for (int i=18;i>=0;i--)
	if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
	
	if (x!=y) x=fa[x][0];
	bj[xx]++;bj[yy]++;bj[x]--;bj[fa[x][0]]--;
}
void dfs2(int u)
{
	for (int i=head[u];i;i=next[i])
	if (to[i]!=fa[u][0])
	{
		dfs2(to[i]);
		f[u]+=f[to[i]];
	}
	f[u]+=bj[u];
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++) scanf("%d",&a[i]);
	for (int x,y,i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y);add(y,x);
	}
	dfs(a[1],1);
	for (int i=1;i<n;i++) find(a[i],a[i+1]);
	dfs2(a[1]);
	for (int i=2;i<=n;i++) f[a[i]]--;
	for (int i=1;i<=n;i++) printf("%d\n",f[i]);
	
	return 0;
}