[樹鏈剖分+線段樹] bzoj4719: [Noip2016]天天愛跑步【留坑待填】
阿新 • • 發佈:2019-02-11
從s到t
上升時,對於經過的每一個節點i
設經過了t[i]條路徑 即用時t[i]
t[i]=dep[s]-dep[i] 得t[i]+dep[i]=dep[s]
同樣 下降時,對於每一個經過的節點i
t[i]=(dep[i]-dep[lca])+(dep[s]-dep[lca])
得dep[i]-t[i]=2*dep[lca]-dep[s]
當i作為答案時 dep[i]-t[i]為定值
對於每一個權值建一顆動態開點的線段樹
然後對於每一次查詢就查詢就查詢這個權的線段樹
然後這種O(nlog^2n) 而且記憶體巨大的破演算法
居然在bzoj上跑過去了……跑過去了!
然後就在各種2s的OJ上T了……
現在不停課了感覺肝一道題都……好累啊
不搞了不搞了搞了我一個星期快死了
留坑待填
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
struct trnode
{
int lc,rc,c,lazy;
}tr[24000000];int trlen;
struct node
{
int x,y,next;
}a[610000];
int last[310000],len;
int fa[310000][25],dep[310000],son[310000],tot[310000],bin[25];
int ys[310000],id,top[310000 ];
int rt[1210000];
int n,m;
void build(int x,int y)
{
len++;
a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;
}
void pre_node(int x)
{
son[x]=0;tot[x]=1;
for(int i=1;i<=20;i++)fa[x][i]=fa[fa[x][i-1]][i-1];
for(int k=last[x];k;k=a[k].next)
{
int y=a[k].y;
if (y!=fa[x][0])
{
fa[y][0]=x;
dep[y]=dep[x]+1;
pre_node(y);
if(tot[son[x]]<tot[y]) son[x]=y;
tot[x]+=tot[y];
}
}
}
int lca(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int i=20;i>=0;i--) if(bin[i]<=dep[x]&&dep[fa[x][i]]>=dep[y]) x=fa[x][i];//不能跳過y
if(x==y) return x;
for(int i=20;i>=0;i--) if(bin[i]<=dep[x]&&fa[x][i]!=fa[y][i]) {x=fa[x][i];y=fa[y][i];}//跳到他們的最近公共祖先的兒子
return fa[x][0];
}
void pre_eage(int x,int tp)
{
ys[x]=++id; top[x]=tp;
if (son[x]!=0) pre_eage(son[x],tp);
for (int k=last[x];k;k=a[k].next)
{
int y=a[k].y;
if (a[k].y!=fa[x][0]&&a[k].y!=son[x])
{
pre_eage(a[k].y,a[k].y);
}
}
}
void update(int x)
{
int lc=tr[x].lc,rc=tr[x].rc;
if (lc==0) lc=tr[x].lc=++trlen;
tr[lc].lazy+=tr[x].lazy;tr[lc].c+=tr[x].lazy;
if (rc==0) rc=tr[x].rc=++trlen;
tr[rc].lazy+=tr[x].lazy;tr[rc].c+=tr[x].lazy;
tr[x].lazy=0;
}
void change(int &x,int xl,int xr,int l,int r,int c)
{
if (x==0) x=++trlen;//動態開點!
if (xl==l&&xr==r)
{
tr[x].c+=c;tr[x].lazy+=c;
return ;
}
int mid=(xl+xr)/2;
if (tr[x].lazy) update(x);
if (r<=mid) change(tr[x].lc,xl,mid,l,r,c);
else if (l>mid) change(tr[x].rc,mid+1,xr,l,r,c);
else change(tr[x].lc,xl,mid,l,mid,c),change(tr[x].rc,mid+1,xr,mid+1,r,c);
}
int getsum(int x,int xl,int xr,int p)
{
if (x==0) return 0;
if (xl==xr) return tr[x].c;
int mid=(xl+xr)/2;
if (tr[x].lazy) update(x);
if (p<=mid) return getsum(tr[x].lc,xl,mid,p);
else return getsum(tr[x].rc,mid+1,xr,p);
}
void solve(int x,int y,int u)
{
int tx=top[x],ty=top[y];
while(tx!=ty)
{
if(dep[tx]>dep[ty]) {swap(tx,ty);swap(x,y);}
change(rt[u],1,n,ys[ty],ys[y],1);
y=fa[ty][0]; ty=top[y];
}
if(x==y) change(rt[u],1,n,ys[x],ys[x],1);
else
{
if(dep[x]>dep[y]) swap(x,y);
change(rt[u],1,n,ys[x],ys[y],1);
}
}
int w[310000];
int main()
{
bin[0]=1;
for (int i=1;i<=20;i++) bin[i]=(bin[i-1]<<1);
scanf("%d%d",&n,&m);
len=0;memset(last,0,sizeof(last));
for (int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
build(x,y);
build(y,x);
}
fa[1][0]=0;dep[1]=1;
pre_node(1);
id=0;pre_eage(1,1);
for (int i=1;i<=n;i++) scanf("%d",&w[i]);
for (int i=1;i<=m;i++)
{
int s,t;
scanf("%d%d",&s,&t);
int ll=lca(s,t);
solve(s,ll,dep[s]);
solve(t,ll,2*dep[ll]-dep[s]+600000);
change(rt[dep[s]],1,n,ys[ll],ys[ll],-1);//lca多走了一次
}
for (int i=1;i<=n;i++)
printf("%d ",getsum(rt[dep[i]+w[i]],1,n,ys[i])+getsum(rt[dep[i]-w[i]+600000],1,n,ys[i]));//上升/下降時剛好能夠到達這個點的定值 (有可能為負數)
printf("\n");
return 0;
}