看大家清一色都是树剖+树上差分的做法,本弱贡献一种用树剖+线段树的思路。
题意很明显 每次把 ui−1−>lca(ui−1,ui)−>ui 路径上的点权值都+1
很明显就是线段树嘛。
我们把所有的区间修改操作都搞好了以后,然后把整颗线段树都遍历一遍,求出每个点上的权值。
但问题是 中间的点每次加都多加了1(不包括第一次的点和最后一次的点)
那我们就用个数组记录下来 最后输出的时候减去多加的值就行。
总耗时1000ms左右 毕竟不是正解思路,也许改成ZKW式线段树会更快
总行数140行左右,比差分多了不少,我真是菜。
#include
#include
#define lson (o<<1),l,mid
#define rson (o<<1)+1,mid+1,r
using namespace std;
const int maxm=310000;
int deep[maxm],son[maxm],size[maxm],fa[maxm],top[maxm];
int head[maxm],to[maxm<<1],net[maxm<<1],cnt;
int p[maxm],val[maxm],who[maxm],ans[maxm],sx[maxm];
int id[maxm],tot;
int n,m,root;
struct node{
int l,r,siz,maxi,sum,add;
};
node t[maxm*4];
struct war{
int u,v;
};
war a[maxm];
inline void add(int x,int y)
{
to[++cnt]=y;
net[cnt]=head[x];
head[x]=cnt;
}
int dfs1(int now,int fax,int dep)
{
fa[now]=fax,deep[now]=dep,size[now]=1;
int maxson=-1;
for(int i=head[now];i;i=net[i])
if(to[i]!=fax)
{
size[now]+=dfs1(to[i],now,dep+1);
if(maxsonreturn size[now];
}
void dfs2(int now,int topx)
{
id[now]=++tot;
who[tot]=now;
val[tot]=p[now];
top[now]=topx;
if(!son[now]) return;
dfs2(son[now],topx);
for(int i=head[now];i;i=net[i])
if(!id[to[i]])
dfs2(to[i],to[i]);
}
inline void update(int o)
{
t[o].sum=(t[(o<<1)].sum+t[(o<<1)|1].sum);
t[o].maxi=max(t[(o<<1)].maxi,t[(o<<1)|1].maxi);
}
inline void pushdown(int o)
{
int adi=t[o].add;
for(int i=0;i<=1;i++)
t[(o<<1)+i].sum=(t[(o<<1)+i].sum+(t[(o<<1)+i].siz*adi)),t[(o<<1)+i].add=(t[(o<<1)+i].add+adi);
t[o].add=0;
}
void build(int o,int l,int r)
{
t[o].l=l,t[o].r=r,t[o].siz=r-l+1;
if(l==r)
{
t[o].sum=t[o].maxi=val[l];
return;
}
int mid=(l+r)>>1;
build(lson),build(rson);
update(o);
}
void adj(int o,int ql,int qr,int num)
{
int l=t[o].l,r=t[o].r;
if(ql<=l&&qr>=r)
{
t[o].sum=(t[o].sum+(t[o].siz*num));
t[o].add=(t[o].add+num);
return;
}
pushdown(o);
int mid=(l+r)>>1;
if(ql<=mid) adj((o<<1),ql,qr,num);
if(qr>mid) adj((o<<1)|1,ql,qr,num);
update(o);
}
inline void tree_x(int u,int v,int num)
{
while(top[u]!=top[v])
{
if(deep[top[u]]1,id[top[u]],id[u],num);
u=fa[top[u]];
}
if(deep[u]1,id[v],id[u],num);
}
inline int read()
{
int x=0,w=1;char ch=0;
while(ch<'0'||ch>'9')
{
if(ch=='-') w=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
return x*w;
}
void print(int o,int l,int r)
{
if(l==r)
{
ans[who[l]]=t[o].sum;
return;
}
int mid=(l+r)>>1;
pushdown(o);
print((o<<1),l,mid);
print((o<<1)|1,mid+1,r);
}
int main()
{
n=read();
for(int i=1;i<=n;i++)
{
a[i].u=read();
if(i!=1) sx[a[i].u]++;
}
for(int i=1,u,v;i1,0,0),dfs2(1,1);
build(1,1,tot);
for(int i=2;i<=n;i++)
tree_x(a[i-1].u,a[i].u,1);
print(1,1,tot);
for(int i=1;i<=n;i++)
printf("%d\n",ans[i]-sx[i]);
return 0;
}