首页 技术 正文
技术 2022年11月11日
0 收藏 899 点赞 2,450 浏览 1825 个字
//第一次A掉紫题就来写题解,我是不是疯了
//说实话这道题还是比较裸的树上差分
//对于树上的一条路径(s,t),我们只需要把ch[s]++,ch[t]++,ch[LCA(S,T)]--,再把lca的爸爸减一
//再dfs跑一遍就可以了
//但这题还是有些不一样的,这道题里的路径终点会算作下一条路径的起点
//解法与楼下的大佬们有些类似,但我可能写的比较好理解一些吧(其实只因为我太弱了)
//还是上代码吧
#include<iostream>
#include<cstring>
#include<fstream>
#include<cstdio>
using namespace std;
const int maxn=300001;
struct node
{
int u,v;
int nxt;
}edge[maxn*2];// 无向边,领接表开两倍
int deep[maxn],ch[maxn],head[maxn],a[maxn];
int f[maxn][30];//倍增数组,f[i][j]表示i向上跳2^j到达的点
int n,num=1;
inline void read(int &x)//读入优化
{
char c=getchar();
x=0;
while(c<'0'||c>'9') c=getchar();
while(c>='0'&&c<='9')
{
x=x*10+c-48;
c=getchar();
}
}
void add_edge(int x,int y)//邻接表加边
{
edge[num].u=x;
edge[num].v=y;
edge[num].nxt=head[x];
head[x]=num;
num++;
}
void build(int r)//建树
{
for(int i=head[r];i!=-1;i=edge[i].nxt)
{
int xx=edge[i].v;
if(deep[xx]==0)//如果点xx没有被访问过的话
{
deep[xx]=deep[r]+1;
f[xx][0]=r;
build(xx);//继续以xx为根建树
}
}
}
void fill()
{
for(int i=1;i<=29;i++)
for(int j=1;j<=n;j++)
f[j][i]=f[f[j][i-1]][i-1];//第j个节点,向上跳i能到达的节点先跳到2^(i-1)处再向上跳2^(i-1)能到达的节点
}
int lca(int x,int y)
{
if(deep[x]<deep[y]) swap(x,y);//如果x在y上面,交换让x往上跳
for(int i=29;i>=0;i--)
if(deep[f[x][i]]>=deep[y])
x=f[x][i];//使x跳到与y同一深度
if(x==y) return x;
for(int i=29;i>=0;i--)
if(f[x][i]!=f[y][i])
{
x=f[x][i];
y=f[y][i];
}
return f[x][0];//再跳一步,找到lca
}
void dfs(int r)
{
for(int i=head[r];i!=-1;i=edge[i].nxt)
{
int xx=edge[i].v;
if(xx==f[r][0]) continue;
dfs(xx);
ch[r]+=ch[xx];//求子树和
}
}
inline void write(int x)//闲的没事干加的输出优化
{
if(x>9) write(x/10);
putchar(x%10+'0');
}
int main()
{
read(n);
for(int i=1;i<=n;i++)
{
read(a[i]);
head[i]=-1;
}
int xx,yy;
for(int i=1;i<=n-1;i++)
{
read(xx);
read(yy);
add_edge(xx,yy);
add_edge(yy,xx);
}
deep[1]=1;
build(1);//以1为根建树
fill();
for(int i=1;i<=n-1;i++)
{
xx=a[i];
yy=a[i+1];
ch[xx]++;
ch[yy]++;
ch[lca(xx,yy)]--;
ch[f[lca(xx,yy)][0]]--;
}
dfs(1);
for(int i=2;i<=n;i++)
ch[a[i]]--;//因为我们把既作为起点又做为终点的点算了两次,现在把这些点的权值减一即可
for(int i=1;i<=n;i++)
{
write(ch[i]);
putchar(char(10));//char(10)即换行
}
return 0;
}
相关推荐
python开发_常用的python模块及安装方法
adodb:我们领导推荐的数据库连接组件bsddb3:BerkeleyDB的连接组件Cheetah-1.0:我比较喜欢这个版本的cheeta…
日期:2022-11-24 点赞:878 阅读:9,498
Educational Codeforces Round 11 C. Hard Process 二分
C. Hard Process题目连接:http://www.codeforces.com/contest/660/problem/CDes…
日期:2022-11-24 点赞:807 阅读:5,911
下载Ubuntn 17.04 内核源代码
zengkefu@server1:/usr/src$ uname -aLinux server1 4.10.0-19-generic #21…
日期:2022-11-24 点赞:569 阅读:6,745
可用Active Desktop Calendar V7.86 注册码序列号
可用Active Desktop Calendar V7.86 注册码序列号Name: www.greendown.cn Code: &nb…
日期:2022-11-24 点赞:733 阅读:6,502
Android调用系统相机、自定义相机、处理大图片
Android调用系统相机和自定义相机实例本博文主要是介绍了android上使用相机进行拍照并显示的两种方式,并且由于涉及到要把拍到的照片显…
日期:2022-11-24 点赞:512 阅读:8,141
Struts的使用
一、Struts2的获取  Struts的官方网站为:http://struts.apache.org/  下载完Struts2的jar包,…
日期:2022-11-24 点赞:671 阅读:5,304