关注
第二个换根dp
A的代码:#include<bits/stdc++.h>
(35927)#define int long long
using namespace std;
const int N = 1e5+5;
struct Node{
int u,v,next;
}a[N];
int n,m;
int tot,last[N];
int ans[N];
int dp[N];
int sum[N];
void add(int u,int v)
{
tot++;
a[tot].u = u;
a[tot].v = v;
a[tot].next = last[u];
last[u] = tot;
}
void dfs(int u,int fa)
{
for(int i = last[u];i >= 0;i = a[i].next)
{
int v = a[i].v;
if(v == fa)
{
continue;
}
dfs(v,u);
sum[u] += sum[v] + 1;
dp[u] += dp[v] + sum[v]+1;
}
}
void dfs2(int u,int fa)
{
for(int i = last[u]; i >= 0; i = a[i].next)
{
int v = a[i].v;
if(v == fa)
{
continue;
}
ans[v] = dp[v] + (ans[u] - (dp[v] + sum[v] + 1 )) + ( n - sum[v] - 1);
dfs2(v,u);
}
}
signed main()
{
tot = -1;
memset(last,-1,sizeof(last));
cin>>n>>m;
for(int i = 2;i <= n; i++)
{
int u,v;
cin>>u;
add(i,u);
add(u,i);
}
memset(dp,0,sizeof(dp));
dfs(1,-1);
ans[1] = dp[1];
dfs2(1,-1);
for(int i = 1;i <= m; i++)
{
int u;
cin>>u;
cout<<ans[u];
if(i == m)
{
cout<<endl;
}
else{
cout<<" ";
}
}
return 0;
}
查看原帖
1 评论
相关推荐
09-04 20:39
南京林业大学 机械工程师 
点赞 评论 收藏
分享

点赞 评论 收藏
分享
牛客热帖
更多
- 1... HR面,到底该准备些啥(附核心问题回答思路)1.6W
- 2... 双非放弃保研,后悔爆哭5719
- 3... 除了卷大厂,还有其他出路吗。。。4132
- 4... 懂车帝二面 2025.10.11 1h32min3462
- 5... 牛牛求救🆘,不敢梭哈后端第二技能点怎么搭配2699
- 6... 双非秋招timeline供参考(腾讯字节阿里快手美团)2668
- 7... 小红书一面面经2440
- 8... 10.12pdd笔试大鸭蛋2168
- 9... 10.12 拼多多技术岗笔试 第二题 求教1923
- 10... 第一次去北京那么远的地方实习,心里总是不安,大家会有这种感觉吗?1811
正在热议
更多
# 面包vs爱情,怎么选? #
6347次浏览 81人参与
# 职场新人体验 #
83188次浏览 590人参与
# 爱玛科技集团求职进展汇总 #
26538次浏览 194人参与
# 实习生如何通过转正 #
103896次浏览 1394人参与
# tplink提前批进度交流 #
206660次浏览 1506人参与
# 安克创新求职进展汇总 #
53575次浏览 528人参与
# 深信服秋招来了 #
279366次浏览 2915人参与
# Tplink求职进展汇总 #
179966次浏览 912人参与
# 秋招结束之后的日子 #
85814次浏览 976人参与
# 面试被问“你的缺点是什么?”怎么答 #
154200次浏览 2137人参与
# 贝壳求职进展汇总 #
34136次浏览 184人参与
# 硬件/芯片公司岗位评价 #
7992次浏览 28人参与
# Offer比较,你最看重什么? #
214800次浏览 1387人参与
# 互联网公司爆料 #
144359次浏览 708人参与
# 招银网络求职进展汇总 #
167594次浏览 991人参与
# 联影求职进展汇总 #
42716次浏览 283人参与
# 华为海思工作体验 #
28789次浏览 120人参与
# 新凯来求职进展汇总 #
49278次浏览 125人参与
# 材料进Fab厂真的劝退吗? #
55811次浏览 204人参与
# 五一之后,实习真的很难找吗? #
87762次浏览 556人参与
# 应届生,你找到工作了吗 #
68696次浏览 459人参与
# 总结:哪家公司最喜欢泡池子 #
143754次浏览 520人参与