题解:P10722 [GESP202406 六级] 二叉树
思路
朴素做法
当输入 \(a_i\) 后,直接将它及它的子树进行变换。而这样时间会超时。预计得分 \(40\) pts。
正解
统计每次变换的节点编号,第 \(i\) 个节点作为根节点进行子树变换的次数为 \(rec_i\)。最后从这棵树的根节点 \(1\) 开始向下 dfs,则每个节点变换的次数为
\[rec_i+k_j
\]
其中 \(k_j\) 为它的父节点的变换次数。
最后输出答案。因为 dfs 只会遍历 \(n\) 个节点,所以时间复杂度为 \(O(n)\),不会超时。
省流:类似于线段树的懒标记。
AC code
#include<bits/stdc++.h>
#define MAX_ 100005
using namespace std;
int n,q,tmp;
string bw;
struct node{
bool color;
int leftson;
int rightson;
};
node tree[MAX_];
int rec[MAX_];
void dfs(int x,int k){
int cnt=k+rec[x];
if(cnt&1) tree[x].color=!tree[x].color;
if(tree[x].leftson) dfs(tree[x].leftson,cnt);
if(tree[x].rightson) dfs(tree[x].rightson,cnt);
}
int main(){
ios::sync_with_stdio(false);
cin>>n;
for(int i=1;i<=n-1;i++){
cin>>tmp;
if(tree[tmp].leftson!=0) tree[tmp].rightson=i+1;
else tree[tmp].leftson=i+1;
}
cin>>bw;
int len=bw.size();
for(int i=0;i<len;i++){
tree[i+1].color=(bw[i]-'0')&1;
}
cin>>q;
for(int i=1;i<=q;i++){
cin>>tmp;
rec[tmp]++;
}
dfs(1,0);
for(int i=1;i<=n;i++){
cout<<tree[i].color;
}
return 0;
}
分类:
题解 / 洛谷 题解
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话