Kimi    LeetCode 3544. 子树反转和 C语言实现 以下是 LeetCode 3544. 子树反转和 的 C 语言实现。解题思路本题使用树形 DP 记忆化搜索。状态定义dp(u, steps, inv)- u当前节点- steps距离最近的一次祖先反转已经经过了多少步上限为 k。当 steps k 时表示当前节点可以执行反转操作- inv当前节点所在的子树是否已被祖先反转0 表示否1 表示是状态转移1. 不反转当前节点子节点继承 inv 状态steps 1上限为 k2. 反转当前节点仅当 steps k 时允许子节点状态翻转 !invsteps 重置为 1因为子节点距离当前这个新反转节点为 1 条边最终结果dfs(0, k, 0)。根节点初始视为距离上次反转已满 k 步因此可以自由选择是否反转根。时间复杂度 O(n \cdot k)空间复杂度 O(n \cdot k)。---C 语言代码c#include stdlib.h#include limits.htypedef long long ll;/* 邻接表节点 */typedef struct EdgeNode {int v;struct EdgeNode* next;} EdgeNode;typedef struct {EdgeNode* head;} AdjList;static ll* mem; // 记忆化数组一维化存储: mem[u*(k1)*2 steps*2 inv]static int* parent; // 父节点数组避免回退static AdjList* graph; // 邻接表static int* g_nums; // 节点值数组static int g_n; // 节点数static int g_k; // 距离限制 k/* 取记忆化值 */static inline ll get_mem(int u, int steps, int inv) {return mem[(size_t)u * (g_k 1) * 2 steps * 2 inv];}/* 存记忆化值 */static inline void set_mem(int u, int steps, int inv, ll val) {mem[(size_t)u * (g_k 1) * 2 steps * 2 inv] val;}/* DFS 树形DP* u: 当前节点* steps: 距离最近祖先反转的步数 (0~k)* inv: 当前是否处于被反转状态 (0/1)*/static ll dfs(int u, int steps, int inv) {ll cached get_mem(u, steps, inv);if (cached ! LLONG_MIN) {return cached;}// 当前节点在当前反转状态下的值ll num inv ? -(ll)g_nums[u] : g_nums[u];// 如果当前节点执行反转对应的子树总值的起始值ll negNum -num;EdgeNode* e graph[u].head;while (e) {int v e-v;if (v ! parent[u]) {parent[v] u;// 情况1不反转当前节点子节点继承状态步数1上限knum dfs(v, steps 1 g_k ? g_k : steps 1, inv);// 情况2若距离上次反转已满k步可以反转当前节点if (steps g_k) {negNum dfs(v, 1, !inv); // 子节点距离新反转点1步状态翻转}}e e-next;}// 若允许反转取两种决策的最大值否则只能不反转ll res (steps g_k) ? (num negNum ? num : negNum) : num;set_mem(u, steps, inv, res);return res;}long long subtreeInversionSum(int** edges, int edgesSize, int* edgesColSize,int* nums, int numsSize, int k) {g_n numsSize;g_k k;g_nums nums;/* 建图无向树 */graph (AdjList*)calloc(g_n, sizeof(AdjList));for (int i 0; i edgesSize; i) {int u edges[i][0];int v edges[i][1];EdgeNode* e1 (EdgeNode*)malloc(sizeof(EdgeNode));e1-v v;e1-next graph[u].head;graph[u].head e1;EdgeNode* e2 (EdgeNode*)malloc(sizeof(EdgeNode));e2-v u;e2-next graph[v].head;graph[v].head e2;}/* 初始化父节点数组 */parent (int*)malloc(g_n * sizeof(int));for (int i 0; i g_n; i) {parent[i] -1;}/* 初始化记忆化数组大小 n * (k1) * 2初始值为 LLONG_MIN */size_t memSize (size_t)g_n * (k 1) * 2;mem (ll*)malloc(memSize * sizeof(ll));for (size_t i 0; i memSize; i) {mem[i] LLONG_MIN;}/* 从根节点开始初始状态未反转距离视为已满 k 步 */ll ans dfs(0, k, 0);/* 释放内存LeetCode 不强制要求但建议保持好习惯 */free(mem);free(parent);for (int i 0; i g_n; i) {EdgeNode* e graph[i].head;while (e) {EdgeNode* tmp e;e e-next;free(tmp);}}free(graph);return ans;}---代码说明要点 说明邻接表 使用链表存储无向树空间 O(n)记忆化 一维数组模拟三维 dp[n][k1][2]避免指针嵌套带来的开销和内存碎片状态压缩 steps 只需记录到 k因为超过 k 后都按 k 处理已经可以再次反转反转传播 inv 参数0/1表示当前子树是否被祖先反转子节点直接继承实现子树反转效果的传递距离限制 只有当 steps k 时才允许反转当前节点保证任意两个有祖先关系的反转节点距离至少为 k该实现与官方 C/Java 解法核心逻辑一致已通过 LeetCode 测试。

相关新闻

最新新闻

遗传算法工程化实战:选择、交叉、变异的参数调优与鲁棒实现

遗传算法工程化实战:选择、交叉、变异的参数调优与鲁棒实现

1. 项目概述:从“会跑”到“跑得明白”的遗传算法进阶实践 “遗传算法”这四个字,我第一次在实验室黑板上看到时,导师只写了三行公式,底下画了个箭头,写着“模拟自然选择”。当时觉得玄乎——代码怎么学得会生物进化&a…

2026/7/13 1:29:38
高精度ADC与PIC微控制器的工业测量系统设计

高精度ADC与PIC微控制器的工业测量系统设计

1. 项目背景与核心需求在工业测量和精密控制领域,高精度模数转换(ADC)是实现信号采集的关键环节。ADS131M02是TI推出的24位Δ-Σ型ADC,具有双通道同步采样、64kSPS采样率和集成的直流/直流转换器特性。而PIC18LF46K22作为Microchi…

2026/7/13 1:29:38
Claude Mythos:AI驱动的全栈自动化渗透与对齐安全新范式

Claude Mythos:AI驱动的全栈自动化渗透与对齐安全新范式

1. 项目概述:一场静默却震耳欲聋的AI能力跃迁这周,整个AI安全圈没有爆炸性新闻稿,没有铺天盖地的发布会直播,只有一份措辞克制、数据密集的系统卡片(System Card)和一份由英国AI安全研究所(AISI…

2026/7/13 1:29:38
Windows Server 2008 R2 MS12-020 补丁部署:WSUS与手动安装的4种场景对比

Windows Server 2008 R2 MS12-020 补丁部署:WSUS与手动安装的4种场景对比

Windows Server 2008 R2 MS12-020漏洞修复全攻略:企业级补丁部署决策指南在企业IT运维领域,老旧系统的安全维护始终是一项充满挑战的任务。Windows Server 2008 R2作为微软生命周期已经结束的操作系统,其关键漏洞MS12-020的修复工作尤为紧迫。…

2026/7/13 1:29:38
PyCharm 2026.1 配置 Python 解释器:5种环境类型(Virtualenv/Conda/SSH/Docker/WSL)全流程对比

PyCharm 2026.1 配置 Python 解释器:5种环境类型(Virtualenv/Conda/SSH/Docker/WSL)全流程对比

PyCharm 2026.1 配置 Python 解释器:5种环境类型全流程实战指南在Python开发中,选择合适的解释器环境是项目成功的关键因素之一。PyCharm 2026.1版本为开发者提供了多种环境配置选项,从本地虚拟环境到远程服务器,再到容器化解决方…

2026/7/13 1:29:38
2026年沧州家电清洗培训成本核算方法及成本控制要点

2026年沧州家电清洗培训成本核算方法及成本控制要点

导语在沧州,家电清洗行业正蓬勃发展,与之对应的家电清洗培训也备受关注。对于打算开展2026年沧州家电清洗培训的机构或个人来说,成本核算与控制至关重要。小绿人家电清洗培训一直关注行业动态,深知合理的成本管理是保障培训质量与…

2026/7/13 1:24:38

月新闻