唯一能得分的题也被自己搞炸了,好的。
考场上读完题基本认定和lca脱不了干系,想了一会确认是树剖。
那么问题来了,考前一节课刚发现自己之前打的树剖是错的。
而且就算是错的我也没信心考场调出来。
于是打了个自认为复杂度不高的暴力,T出天际,被各位不屑于打树剖的巨佬の向上标记干趴下。
考完翻书学了一下树上差分,赶脚不错。
树剖||Tarjan求lca+树上差分
后者的思路是预处理路线上每两个房间的lca
每次num[x]++,num[y]++,num[lca]++,num[father[lca]]++
最后进行一次dfs统计所有点的权值
int dfs(int x)
{
v[x]=1;int now=0;
for(int i=head[x];i;i=nxt[i])
{
int y=to[i];
if(v[y])continue;
now+=dfs(y);
}
num[x]+=now;
return num[x];
}
最后num[way[1]]++
然后输出way[i]-1 (因为除了第一个点之外的每个点都算了两遍)
#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;
const int N=300010;
int n;
int to[N<<1],nxt[N<<1],num[N],tot=0,fa[N],v[N],root,head[N],ans[N],way[N],father[N];
vector<int> q[N],qi[N];
void add(int x,int y)
{
to[++tot]=y;
nxt[tot]=head[x];
head[x]=tot;
}
void insq(int x,int y,int i)
{
q[x].push_back(y);
qi[x].push_back(i);
q[y].push_back(x);
qi[y].push_back(i);
}
int getf(int x)
{
if(x==fa[x])return x;
fa[x]=getf(fa[x]);
return fa[x];
}
void tarjan(int x)
{
v[x]=1;
for(int i=head[x];i;i=nxt[i])
{
int y=to[i];
if(v[y])continue;
tarjan(y);
fa[y]=father[y]=x;
}
for(int i=0;i<q[x].size();i++)
{
int y=q[x][i],id=qi[x][i];
if(v[y]==2)ans[id]=getf(y);
}
v[x]=2;
}
int dfs(int x)
{
v[x]=1;int now=0;
for(int i=head[x];i;i=nxt[i])
{
int y=to[i];
if(v[y])continue;
now+=dfs(y);
}
num[x]+=now;
return num[x];
}
int main()
{
/*freopen("FU.in","r",stdin);
freopen("FU.out","w",stdout);*/
scanf("%d",&n);
for(int i=1;i<=n;i++)scanf("%d",&way[i]),fa[i]=i;
root=way[1];
for(int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
add(x,y);add(y,x);
x=way[i],y=way[i+1];
insq(x,y,i);
}
/*cout<<endl;
for(int i=1;i<=n;i++)cout<<father[i]<<' ';
cout<<endl;*/
tarjan(root);
memset(v,0,sizeof(v));
for(int i=1;i<n;i++)
{
int x=way[i],y=way[i+1],lca=ans[i];
num[x]++;num[y]++;num[lca]--;num[father[lca]]--;
}
dfs(root);
num[root]++;
for(int i=1;i<=n;i++)printf("%d\n",num[i]-1);
return 0;
}