没看过正解。。应该是些乱七八糟想不出来的东西
解法1:
首先,必须要做的是将每条路径拆成2个直的路径
那么对于那条从深度大的到深度小的路径 dep[x]-dep[y]应该等于观察时间
那么就可以在这些点打标记
那问题在于怎么找这些点
可以把深度为x的数组用vector搞出来
然后每次判断一下x里面的元素是否在他的子树中(判断一下lca就好了)
树剖判断一下每个点的覆盖次数
对于另一条路径同理做
代码:
解法2:
这个解法复杂度是可以分析出来的
首先还是把路径拆成两条
那么对于那条从深度大的到深度小的路径
考虑启发式合并
对于每个点维护一颗splay表示子树中出现的dep值的个数
将路径顶端和底端分别在该点打标记
访问到顶端时删除底端时插入
由于每一次向上会导致所有元素的权值变化所以可以加一个标记(插入时减掉这个值)
那么到查询点的时候就在splay上查询就可以了
那么复杂度是nlogn*splay的复杂度 也就是nlog^2n 不过splay的常数感觉这方法还是很虚啊。。。
还有就是splay的insert操作之后要splay一下
防止是这颗树的第一个节点
代码:
#include <bits/stdc++.h>
using namespace std;
#define maxn 6200000
#define maxn2 310000
int count2[maxn],leftson[maxn],rightson[maxn],fa[maxn],root[maxn2];
struct re{
int a,b,c;
}a[maxn2*2];
int ans[maxn2],bz[maxn2][20],dep[maxn2],vw[maxn2],head[maxn2],l,last[maxn2];
int hf[maxn2],cc[maxn2],dd[maxn2],lcaa[maxn2],num[maxn],num2,data[maxn];
int n,m;
vector<int> ve[maxn];
inline void updata(int x)
{
count2[x]=count2[leftson[x]]+count2[rightson[x]]+1;
}
inline void rotate(int x,int y)
{
int father=fa[x];
if (y==1)
{
rightson[father]=leftson[x];
if (leftson[x]) fa[leftson[x]]=father;
} else
{
leftson[father]=rightson[x];
if (rightson[x]) fa[rightson[x]]=father;
}
fa[x]=fa[father];
if (fa[father])
{
if (leftson[fa[father]]==father) leftson[fa[father]]=x;
else rightson[fa[father]]=x;
}
fa[father]=x;
if (y==1) leftson[x]=father; else rightson[x]=father;
updata(father); updata(x);
}
inline void splay(int k,int x,int goal)
{
if (x==root[k]) return;
int father=fa[x];
while (father!=goal)
{
if (fa[father]==goal)
{
if (x==leftson[father]) rotate(x,2); else rotate(x,1);
} else
{
if (father==leftson[fa[father]])
{
if (x==leftson[father]) rotate(father,2),rotate(x,2);
else rotate(x,1),rotate(x,2);
} else
{
if (x==rightson[father]) rotate(father,1),rotate(x,1);
else rotate(x,2),rotate(x,1);
}
}
father=fa[x];
}
if (goal==0) root[k]=x;
}
inline void dfs(int x,int y)
{
bz[x][0]=y; dep[x]=dep[y]+1;
int u=head[x];
while (u)
{
int v=a[u].b;
if (v!=y) dfs(v,x);
u=a[u].a;
}
}
inline int get_lca(int x,int y)
{
if (dep[x]<dep[y]) swap(x,y);
for (int i=19;i>=0;i--)
if (dep[bz[x][i]]>=dep[y]) x=bz[x][i];
if (x==y) return(x);
for (int i=19;i>=0;i--)
if (bz[x][i]!=bz[y][i])
{
x=bz[x][i],y=bz[y][i];
}
return(bz[x][0]);
}
inline void arr(int x,int y)
{
a[++l].a=head[x];
a[l].b=y;
head[x]=l;
}
inline void insert(int x,int v,int w)
{
int tmp=x;x=root[x];
while (x)
{
if (data[x]==v) break;
count2[x]++;
if (v<data[x])
{
if (!leftson[x]) break;
x=leftson[x];
} else
{
if (!rightson[x]) break;
x=rightson[x];
}
}
if (x!=0&&data[x]==v) num[x]+=w; else
{
data[++num2]=v; count2[num2]=1; fa[num2]=x;
num[num2]=w;
if (x!=0)
{
if (v>data[x]) rightson[x]=num2;
else leftson[x]=num2;
}
splay(tmp,num2,0);
}
}
inline void delete1(int k)
{
int x=leftson[root[k]];
if (x==0)
{
root[k]=rightson[root[k]]; fa[root[k]]=0; return;
}
while (rightson[x]) x=rightson[x];
splay(k,x,root[k]);
rightson[x]=rightson[root[k]];
if (rightson[root[k]]) fa[rightson[root[k]]]=x;
updata(x);
fa[x]=0; root[k]=x;
}
inline int search(int x,int goal)
{
x=root[x];
while (1)
{
if (data[x]==goal) return(x);
if (data[x]<goal)
{
if (!rightson[x]) return(x);
x=rightson[x];
}
else
{
if (!leftson[x]) return(x);
x=leftson[x];
}
}
}
inline void merge(int x,int y,int z)
{
while (root[x])
{
int tmp=data[root[x]]-last[y]+last[x];
int o=search(y,tmp);
if (o!=0&&data[o]==tmp) num[o]+=num[root[x]];
else insert(y,tmp,num[root[x]]);
delete1(x);
}
root[z]=root[y]; last[z]=last[y];
}
inline void dfs2(int x,int fa,int lq)
{
int u=head[x],pp=ve[x].size();
for (int i=0;i<=pp/2-1;i++)
if (!ve[x][i*2+1])
{
insert(x,ve[x][i*2],1);
}
while (u)
{
int v=a[u].b;
if (v!=fa)
{
dfs2(v,x,lq);
last[v]+=lq;
if (count2[root[x]]<count2[root[v]]) merge(x,v,x);
else merge(v,x,x);
}
u=a[u].a;
}
int tmp=search(x,vw[x]-last[x]);
if (data[tmp]==vw[x]-last[x]) ans[x]+=num[tmp];
for (int i=0;i<=pp/2-1;i++)
if (ve[x][i*2+1])
{
int tmp=search(x,ve[x][i*2]-last[x]);
if (num[tmp]>1) num[tmp]--;
else
{
splay(x,tmp,0);
delete1(x);
}
}
}
int main()
{
freopen("noip.in","r",stdin);
freopen("noip.out","w",stdout);
std::ios::sync_with_stdio(false);
cin>>n>>m;
int c,d;
for (int i=1;i<=n-1;i++)
cin>>c>>d,arr(c,d),arr(d,c);
for (int i=1;i<=n;i++)
cin>>vw[i];
dfs(1,0);
for (int i=1;i<=19;i++)
for (int j=1;j<=n;j++)
bz[j][i]=bz[bz[j][i-1]][i-1];
for (int i=1;i<=m;i++)
{
cin>>c>>d;
cc[i]=c; dd[i]=d;
int w=get_lca(c,d);
lcaa[i]=w;
hf[i]=dep[c]-dep[w];
}
for (int i=1;i<=m;i++)
{
ve[cc[i]].push_back(0); ve[cc[i]].push_back(0);
ve[lcaa[i]].push_back(hf[i]); ve[lcaa[i]].push_back(1);
}
dfs2(1,0,1);
memset(leftson,0,sizeof(leftson));
memset(rightson,0,sizeof(rightson));
memset(num,0,sizeof(num));
memset(count2,0,sizeof(count2));
memset(fa,0,sizeof(fa));
memset(root,0,sizeof(root));
memset(last,0,sizeof(last)); num2=0;
memset(data,0,sizeof(data));
for (int i=1;i<=n;i++)
{
vector<int> tmp; swap(tmp,ve[i]);
// ve[i].clear();
}
for (int i=1;i<=m;i++)
{
ve[dd[i]].push_back(hf[i]+dep[dd[i]]-dep[lcaa[i]]);
ve[dd[i]].push_back(0);
ve[lcaa[i]].push_back(hf[i]);
ve[lcaa[i]].push_back(1);
}
dfs2(1,0,-1);
for (int i=1;i<=m;i++)
{
if (hf[i]==vw[lcaa[i]]) ans[lcaa[i]]--;
}
for (int i=1;i<=n-1;i++) cout<<ans[i]<<" ";
cout<<ans[n];
return 0;
}