Slightly rustier get_counts and get_buckets.

This commit is contained in:
Nicolas Patry
2020-07-31 18:23:16 +02:00
parent b4eeec562b
commit 1a4e8cffab

View File

@@ -5,26 +5,30 @@ fn has_high_bit(j: usize) -> bool {
} }
fn get_counts(t: &StringT, c: &mut Bucket) { fn get_counts(t: &StringT, c: &mut Bucket) {
for item in c.iter_mut() { c.iter_mut().map(|c| *c = 0).count();
*item = 0; t.iter()
} .map(|character| c[*character as usize] += 1)
for character in t { .count();
c[*character as usize] += 1;
}
} }
fn get_buckets(c: &Bucket, b: &mut Bucket, k: usize, end: bool) { fn get_buckets(c: &Bucket, b: &mut Bucket, _k: usize, end: bool) {
let mut sum = 0; let mut sum = 0;
if end { if end {
for i in 0..k { b.iter_mut()
sum += c[i]; .enumerate()
b[i] = sum; .map(|(i, b_el)| {
} sum += c[i];
*b_el = sum;
})
.count();
} else { } else {
for i in 0..k { b.iter_mut()
b[i] = sum; .enumerate()
sum += c[i]; .map(|(i, b_el)| {
} *b_el = sum;
sum += c[i];
})
.count();
} }
} }