2022-09-27:给定一个棵树, 树上每个节点都有自己的值,记录在数组nums里, 比如nums[4] = 10,表示4号点的值是10, 给定树上的每一条边,记录在二维数组edges里, 比如ed
2022-09-27:给定一个棵树,
树上每个节点都有自己的值,记录在数组nums里,
比如nums[4] = 10,表示4号点的值是10,
给定树上的每一条边,记录在二维数组edges里,
比如edges[8] = {4, 9}表示4和9之间有一条无向边,
可以保证输入一定是一棵树,只不过边是无向边,
那么我们知道,断掉任意两条边,都可以把整棵树分成3个部分。
假设是三个部分为a、b、c,
a部分的值是:a部分所有点的值异或起来,
b部分的值是:b部分所有点的值异或起来,
c部分的值是:c部分所有点的值异或起来,
请问怎么分割,能让最终的:三个部分中最大的异或值 - 三个部分中最小的异或值,最小。
返回这个最小的差值。
输入:nums = [1,5,5,4,11], edges = [[0,1],[1,2],[1,3],[3,4]]。
输出:9。
答案2022-09-27:
dfn序号。
这道题来自力扣2322。力扣上测试了好几种语言的代码,go语言运行效率是最高,其次是java;rust表现不佳,原因是代码中有复制切片的行为。内存占用go是最低的,rust偏高,原因是代码中有复制切片的行为。
时间复杂度:O(n^2)。
空间复杂度:O(n)。
代码用rust编写。代码如下:
use std::iter::repeat;
fn main() {
let mut nums = vec![1, 5, 5, 4, 11];
let mut edges = vec![vec![0, 1], vec![1, 2], vec![1, 3], vec![3, 4]];
let ans = minimum_score(&mut nums, &mut edges);
println!("ans = {}", ans);
}
const MAX_VALUE: i32 = 1 << 31 - 1;
static mut cnt: i32 = 0;
fn minimum_score(nums: &mut Vec<i32>, edges: &mut Vec<Vec<i32>>) -> i32 {
let n = nums.len() as i32;
// 先建立图
// ArrayList<ArrayList<Integer>> graph = new ArrayList<>();
let mut graph: Vec<Vec<i32>> = vec![];
// 4个点,0、1、2、3
// 0 : {}
// 1 : {}
// 2 : {}
// 3 : {}
for _i in 0..n {
graph.push(vec![]);
}
for edge in edges.iter() {
// a,b
// graph.get(a).add(b);
// graph.get(b).add(a);
graph[edge[0] as usize].push(edge[1]);
graph[edge[1] as usize].push(edge[0]);
}
// 无向边组成的无环图
// 为了方便,就认为0是头
// dfn[i] = ?
let mut dfn: Vec<i32> = repeat(0).take(n as usize).collect();
// xor[i] 以i为头的整棵树,整体异或的结果是多少?
let mut xor: Vec<i32> = repeat(0).take(n as usize).collect();
// size[i] 以i为头的整棵树,一共几个点?
let mut size: Vec<i32> = repeat(0).take(n as usize).collect();
unsafe {
cnt = 1;
}
dfs(nums, &mut graph, 0, &mut dfn, &mut xor, &mut size);
let mut ans = MAX_VALUE;
let m = edges.len() as i32;
let mut cut1: i32;
let mut cut2: i32;
let mut pre: i32;
let mut pos: i32;
let mut part1: i32;
let mut part2: i32;
let mut part3: i32;
let mut max: i32;
let mut min: i32;
for i in 0..m {
// i,要删掉的第一条边,i号边
// edges[i][0] edges[i][1] dfn 谁大,谁就是删掉之后的树的头!cut1
// a b cut1
// { a, b}
// 0 1
let a = edges[i as usize][0];
let b = edges[i as usize][1];
cut1 = if dfn[a as usize] < dfn[b as usize] {
b
} else {
a
};
for j in (i + 1)..m {
// j, 要删掉的第二条边,j号边
// { c, d}
// 0 1
let c = edges[j as usize][0];
let d = edges[j as usize][1];
cut2 = if dfn[c as usize] < dfn[d as usize] {
d
} else {
c
};
// cut1,cut2
pre = if dfn[cut1 as usize] < dfn[cut2 as usize] {
cut1
} else {
cut2
};
pos = if pre == cut1 { cut2 } else { cut1 };
// 早 pre 晚 pos
part1 = xor[pos as usize];
// pos为头的树,是pre为头的树的子树!
if dfn[pos as usize] < dfn[pre as usize] + size[pre as usize] {
part2 = xor[pre as usize] ^ xor[pos as usize];
part3 = xor[0] ^ xor[pre as usize];
} else {
// pos为头的树,不是pre为头的树的子树!
part2 = xor[pre as usize];
part3 = xor[0] ^ part1 ^ part2;
}
max = get_max(get_max(part1, part2), part3);
min = get_min(get_min(part1, part2), part3);
ans = get_min(ans, max - min);
}
}
return ans;
}
fn get_max<T: Clone + Copy + std::cmp::PartialOrd>(a: T, b: T) -> T {
if a > b {
a
} else {
b
}
}
fn get_min<T: Clone + Copy + std::cmp::PartialOrd>(a: T, b: T) -> T {
if a < b {
a
} else {
b
}
}
// 所有节点的值,存在nums数组里
// 整个图结构,存在graph里
// 当前来到的是cur号点
// 请把cur为头,整棵树,所有节点的dfn、size、xor填好!
// 返回!
fn dfs(
nums: &mut Vec<i32>,
graph: &mut Vec<Vec<i32>>,
cur: i32,
dfn: &mut Vec<i32>,
xor: &mut Vec<i32>,
size: &mut Vec<i32>,
) {
// 当前节点了!,
dfn[cur as usize] = unsafe { cnt };
unsafe {
cnt += 1;
}
// 只是来到了cur的头部!
xor[cur as usize] = nums[cur as usize];
size[cur as usize] = 1;
// 遍历所有的孩子!
for next in graph.clone()[cur as usize].iter() {
//有clone,会影响性能
// 只有dfn是0的孩子,才是cur在树中的下级!!!!
if dfn[*next as usize] == 0 {
// cur某个孩子是next
dfs(nums, graph, *next, dfn, xor, size);
// next整棵树的异或和,
xor[cur as usize] ^= xor[*next as usize];
// next整棵树的size
size[cur as usize] += size[*next as usize];
}
}
}
执行结果如下:


2022-09-27:给定一个棵树, 树上每个节点都有自己的值,记录在数组nums里, 比如nums[4] = 10,表示4号点的值是10, 给定树上的每一条边,记录在二维数组edges里, 比如ed的更多相关文章
- hdu6035 Colorful Tree 树形dp 给定一棵树,每个节点有一个颜色值。定义每条路径的值为经过的节点的不同颜色数。求所有路径的值和。
/** 题目:hdu6035 Colorful Tree 链接:http://acm.hdu.edu.cn/showproblem.php?pid=6035 题意:给定一棵树,每个节点有一个颜色值.定 ...
- ytu 1050:写一个函数,使给定的一个二维数组(3×3)转置,即行列互换(水题)
1050: 写一个函数,使给定的一个二维数组(3×3)转置,即行列互换 Time Limit: 1 Sec Memory Limit: 128 MBSubmit: 154 Solved: 112[ ...
- 剑指offer1: 组类型——二维数组中的查找(给定一个数字,查找是否在该数组中)
1. 思路: 缩小范围 2. 方法: (1)要查找的数字等于数组中的数字,结束查找过程: (2)要查找的数字小于数组中的数字,去除该数字右边的数字,在剩下的数字里查找: (3)要查找的数字大于数组中的 ...
- 计算机二级-C语言-程序设计题-190119记录-求出一个二维数组每一列的最小值。
//编写一个函数:tt指向一个M行N列的二维数组,求出二维数组每列中最小的元素,并依次放入pp所指的一维数组中.二维数组中的数在主函数中赋予. //重难点:求出的是每一列的最小值,这里要注意,学会简化 ...
- nRF51800 蓝牙学习 进程记录 2:关于二维数组 执念执战
前天在玩OLED时想完成一直想弄得一个东西,就是简单的单片机游戏.因为STM32和nRF51822的内存足够,所以就用缓存数组的方法来显示图像(我也不知道术语是啥,反正就是在内存中建立一个128X64 ...
- php数组根据某一个键值,把相同键值的合并生成一个新的二维数组
http://blog.csdn.net/xyzchenxiaolin/article/details/51700485 源数据: $infos = array( array( 'a' => 3 ...
- 题目要求:传入数组 内容为[['lisi','男','27'],['wangwu','男',18],['zhaoliu','男','30']],将此二维数组转化为一维数组,创建自定义函数完成
//自定义函数 function getOne($arr){ $str = ''; //定义空的字符串,用来接收值 foreach ($arr as $key=>$value){ //循环遍历数 ...
- PHP如何判断一个数组是一维数组或者是二维数组?用什么函数?
如题:如何判断一个数组是一维数组或者是二维数组?用什么函数? 判断数量即可 <?php if (count($array) == count($array, 1)) { echo '是一维数组' ...
- java怎么定义一个二维数组?
java中使用 [][] 来定义二维数组 定义数组时也可同时初始化下面是一些例子float[][] numthree; //定义一个float类型的2维数组numthree=new float[5][ ...
- PHP如何随机获取一个二维数组中的一个值
获取一个数组: $awardid_list=pdo_fetchall('select id from '.tablename($this->table_award)); 这是微擎的写法哈,意思就 ...
随机推荐
- [Unity移动端]真机调试
一.Android Studio 1.log打印 打开AS,新建一个工程,点击左下角的Logcat,如下图,1是设备(支持模拟器,如果是真机的话,需要usb连接电脑,真机要是usb调试模式),2是包名 ...
- PHP文件及运行(适合PHP初学者)
PHP文件可包含HTML.JavaScript代码和 PHP代码,换句话说PHP 代码可以嵌入HTML文档.PHP文件名以php为后缀. PHP代码以"<?php"开头,以& ...
- Unity图片转存及读取
[code]csharpcode: /// <summary> /// 加载图片 /// </summary> private Sprite LoadTexture(strin ...
- 对利用jsp模板编写登录、注册界面的方法言
使用模板的相关操作步骤详解 1.可以在相关的网站上面找相关的css或者js文件,下载到一个特定的文件夹里面,以备使用 2.然后,将存有相关代码的文件夹直接复制粘贴到web文件下,就会直接保存,可以根据 ...
- 对Javaweb的相关练习之利用.jsp文件和.java文件将输入的数据存储到指定的数据库中
练习分析 import javax.servlet.*; import javax.servlet.annotation.WebServlet; import javax.servlet.http.* ...
- Linux & 标准C语言学习 <DAY2>
vim文本编辑器: 可以直接在终端下采用纯键盘操作的一款文本编辑器,号称编辑器之神,可以二次升级.可以扩展 基础用法: 1.进入vim: 输入 ...
- 声网自研传输层协议 AUT 的落地实践丨Dev for Dev 专栏
本文为「Dev for Dev 专栏」系列内容,作者为声网大后端传输协议负责人 夏天. 针对实时互动应用对网络传输带来的新需求和新挑战,声网通过将实时互动中的应用层业务需求与传输策略的分层和解耦,于 ...
- IDEA-日志管理神器
Grep Console-插件的好处就在于能使控制台输出日志时,可以直接修改插件中定义好的规则,也可以根据自己定义的规则,输出不同的颜色.这样就可以将错误信息标记成显眼的颜色,方便查看,提高bug寻找 ...
- 正则表达式(cases)
匹配不含'aaa'和'bbb'的字符串: ^(?!.*aaa)(?!.*bbb).*$ To be continue...
- 通过 poe 免费使用ChatGPT、GPT-4
poe 是由美版知乎 Quora 构建的AI 产品,提供实时在线与多个AI 机器人交流.Quora 于去年 12 月首次推出Poe 作为封闭测试版,并于2月份向所有 iOS 用户开放.支持 web 端 ...