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;
}