写点什么

2022-09-03:n 块石头放置在二维平面中的一些整数坐标点上 每个坐标点上最多只能有一块石头 如果一块石头的 同行或者同列 上有其他石头存在,那么就可以移除这块石头。 给你一个长度为 n 的数组

  • 2022 年 9 月 03 日
    北京
  • 本文字数:1381 字

    阅读完需:约 5 分钟

2022-09-03:n块石头放置在二维平面中的一些整数坐标点上 每个坐标点上最多只能有一块石头 如果一块石头的 同行或者同列 上有其他石头存在,那么就可以移除这块石头。 给你一个长度为 n 的数组

2022-09-03:n 块石头放置在二维平面中的一些整数坐标点上每个坐标点上最多只能有一块石头如果一块石头的 同行或者同列 上有其他石头存在,那么就可以移除这块石头。给你一个长度为 n 的数组 stones ,其中 stones[i] = [xi, yi] 表示第 i 块石头的位置,返回 可以移除的石子 的最大数量。输入: stones = [[0,0],[0,1],[1,0],[1,2],[2,1],[2,2]]。输出: 5。


答案 2022-09-03:


并查集。行代表和列代表合并。


代码用 rust 编写。代码如下:


use std::collections::HashMap;fn main() {    let mut stones = vec![        vec![0, 0],        vec![0, 1],        vec![1, 0],        vec![1, 2],        vec![2, 1],        vec![2, 2],    ];    let ans = remove_stones(&mut stones);    println!("ans = {}", ans);}
fn remove_stones(stones: &mut Vec<Vec<i32>>) -> i32 { let n = stones.len() as i32; let mut row_pre: HashMap<i32, i32> = HashMap::new(); let mut col_pre: HashMap<i32, i32> = HashMap::new(); let mut uf = UnionFind::new(n); for i in 0..n { let x = stones[i as usize][0]; let y = stones[i as usize][1]; if !row_pre.contains_key(&x) { row_pre.insert(x, i); } else { uf.union(i, *row_pre.get(&x).unwrap()); } if !col_pre.contains_key(&y) { col_pre.insert(y, i); } else { uf.union(i, *col_pre.get(&y).unwrap()); } } return n - uf.sets();}pub struct UnionFind { father: Vec<i32>, size: Vec<i32>, help: Vec<i32>, sets: i32,}impl UnionFind { fn new(n: i32) -> UnionFind { let mut father: Vec<i32> = vec![]; let mut size: Vec<i32> = vec![]; let mut help: Vec<i32> = vec![]; for i in 0..n { father.push(i); size.push(1); help.push(0); } UnionFind { father, size, help, sets: n, } }
fn union(&mut self, mut i: i32, mut j: i32) { i = self.find(i); j = self.find(j); if i != j { if self.size[i as usize] >= self.size[j as usize] { self.father[j as usize] = i; self.size[i as usize] += self.size[j as usize]; } else { self.father[i as usize] = j; self.size[j as usize] += self.size[i as usize]; } self.sets -= 1; } }
fn find(&mut self, mut i: i32) -> i32 { let mut s = 0; while i != self.father[i as usize] { self.help[s] = i; s += 1; i = self.father[i as usize]; } while s > 0 { s -= 1; self.father[self.help[s] as usize] = i; } return i; }
fn sets(&mut self) -> i32 { self.sets }}
复制代码


执行结果如下:





左神java代码

发布于: 刚刚阅读数: 2
用户头像

还未添加个人签名 2021.02.15 加入

还未添加个人简介

评论

发布
暂无评论
2022-09-03:n块石头放置在二维平面中的一些整数坐标点上 每个坐标点上最多只能有一块石头 如果一块石头的 同行或者同列 上有其他石头存在,那么就可以移除这块石头。 给你一个长度为 n 的数组_算法_福大大架构师每日一题_InfoQ写作社区