Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions rustfmt.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
max_width = 119
6 changes: 1 addition & 5 deletions src/collection.rs
Original file line number Diff line number Diff line change
Expand Up @@ -28,11 +28,7 @@ impl<T: Copy> Iterator for Bitset<T> {

pub fn bitset<T: Copy>(a: Vec<T>) -> Bitset<T> {
let len = a.len();
Bitset {
curr: 0,
array: a,
len,
}
Bitset { curr: 0, array: a, len }
}

#[cfg(test)]
Expand Down
6 changes: 1 addition & 5 deletions src/geometry.rs
Original file line number Diff line number Diff line change
Expand Up @@ -179,11 +179,7 @@ pub fn convex_hull(ps: Vec<(i64, i64)>) -> Vec<(i64, i64)> {
k += 1;
}

let mut ret: Vec<(i64, i64)> = deque
.into_iter()
.take(k - 1)
.map(|pair| (pair.0, pair.1))
.collect();
let mut ret: Vec<(i64, i64)> = deque.into_iter().take(k - 1).map(|pair| (pair.0, pair.1)).collect();

ret.sort_unstable();
ret
Expand Down
27 changes: 5 additions & 22 deletions src/graph.rs
Original file line number Diff line number Diff line change
Expand Up @@ -51,10 +51,7 @@ impl Graph {
}

pub fn connect_with_residual(&mut self, from: usize, to: usize, weight: usize) {
assert!(
self.directed,
"connect_with_residual only works in directed graph."
);
assert!(self.directed, "connect_with_residual only works in directed graph.");

self.graph[from].push((to, weight));
self.out_degrees[from] += 1;
Expand Down Expand Up @@ -93,14 +90,7 @@ mod test_graph {
graph.connect(1, 0, 10);
graph.connect(3, 4, 3);

let expected = vec![
vec![(1, 1)],
vec![(0, 10)],
vec![(5, 1)],
vec![(4, 3)],
vec![],
vec![],
];
let expected = vec![vec![(1, 1)], vec![(0, 10)], vec![(5, 1)], vec![(4, 3)], vec![], vec![]];
assert_eq!(graph.graph, expected);

assert!(graph.connected(0, 1));
Expand Down Expand Up @@ -260,7 +250,7 @@ impl DepthFirstSearch {

while let Some(node_type) = stack.pop_front() {
match node_type {
NodeType::Forward(u) => {
NodeType::Forward(u) => {
for &(v, _) in self.graph.graph[u].iter() {
if self.seen[v] {
continue;
Expand Down Expand Up @@ -408,10 +398,7 @@ mod test_dijkstra {

let mut dijkstra = Dijkstra::new(graph);
dijkstra.search(0);
assert_eq!(
dijkstra.dist,
vec![0, 1, 6, 3, 5, 8, 10, Dijkstra::INF, Dijkstra::INF]
);
assert_eq!(dijkstra.dist, vec![0, 1, 6, 3, 5, 8, 10, Dijkstra::INF, Dijkstra::INF]);

assert_eq!(dijkstra.shortest_path(0, 3), vec![(0, 1), (1, 3)]);
assert_eq!(dijkstra.shortest_path(0, 8), vec![]);
Expand Down Expand Up @@ -574,11 +561,7 @@ impl LowestCommonAncestor {
let n = graph.n;
let parents = vec![vec![Self::ROOT; n]; Self::LOGV];
let depth = vec![Self::ROOT; n];
Self {
parents,
depth,
graph,
}
Self { parents, depth, graph }
}

pub fn init(&mut self) {
Expand Down
4 changes: 1 addition & 3 deletions src/math.rs
Original file line number Diff line number Diff line change
Expand Up @@ -249,9 +249,7 @@ impl SequentialPrimeFactorization {
i += 1;
}

Self {
smallest_prime_factors,
}
Self { smallest_prime_factors }
}

pub fn factorize(&self, mut x: usize) -> Vec<usize> {
Expand Down
2 changes: 1 addition & 1 deletion src/search.rs
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ pub fn lower_bound(range: std::ops::Range<usize>, prop: &dyn Fn(usize) -> bool)
}

match ok.eq(&range.end) {
true => None,
true => None,
false => Some(ok),
}
}
Expand Down
44 changes: 18 additions & 26 deletions src/tree.rs
Original file line number Diff line number Diff line change
Expand Up @@ -68,32 +68,6 @@ impl SegmentTree {
ret
}

fn range_query_recursive(
&self,
op: &Op,
ql: usize,
qr: usize,
sl: usize,
sr: usize,
pos: usize,
) -> i64 {
if qr <= sl || sr <= ql {
return SegmentTree::default(op);
}
if ql <= sl && sr <= qr {
return self.data[pos];
}

let sm = (sl + sr) / 2;
let lv = self.range_query_recursive(op, ql, qr, sl, sm, pos * 2);
let rv = self.range_query_recursive(op, ql, qr, sm, sr, pos * 2 + 1);
match op {
Op::Add => lv + rv,
Op::Max => lv.max(rv),
Op::Min => lv.min(rv),
}
}

pub fn get_range(&self, l: usize, r: usize) -> i64 {
if let Mode::RangeGet(op) = &self.mode {
self.range_query_recursive(op, l, r, 0, SegmentTree::SEQ_LEN, 1)
Expand Down Expand Up @@ -154,6 +128,24 @@ impl SegmentTree {
panic!("Mode {:?} is not supported.", &self.mode);
}
}

fn range_query_recursive(&self, op: &Op, ql: usize, qr: usize, sl: usize, sr: usize, pos: usize) -> i64 {
if qr <= sl || sr <= ql {
return SegmentTree::default(op);
}
if ql <= sl && sr <= qr {
return self.data[pos];
}

let sm = (sl + sr) / 2;
let lv = self.range_query_recursive(op, ql, qr, sl, sm, pos * 2);
let rv = self.range_query_recursive(op, ql, qr, sm, sr, pos * 2 + 1);
match op {
Op::Add => lv + rv,
Op::Max => lv.max(rv),
Op::Min => lv.min(rv),
}
}
}

#[cfg(test)]
Expand Down