mirror of
https://gitlab.com/Anson-Projects/anson-stuff/Go-v-Rust-Quicksort.git
synced 2025-06-16 06:06:40 +00:00
read data into rust
This commit is contained in:
parent
a24ad4a2d1
commit
84b0505246
159
rust/src/main.rs
159
rust/src/main.rs
@ -1,156 +1,11 @@
|
||||
use rayon::join;
|
||||
|
||||
fn partition<T, F>(d: &mut [T], is_less: &F) -> usize
|
||||
where
|
||||
F: Fn(&T, &T) -> bool,
|
||||
{
|
||||
d.swap(0, d.len() / 2);
|
||||
let mut mid = 0;
|
||||
for i in 1..d.len() {
|
||||
if is_less(&d[i], &d[0]) {
|
||||
mid += 1;
|
||||
d.swap(i, mid);
|
||||
}
|
||||
}
|
||||
d.swap(0, mid);
|
||||
mid
|
||||
}
|
||||
|
||||
fn insert_sort<T, F>(d: &mut [T], is_less: &F)
|
||||
where
|
||||
F: Fn(&T, &T) -> bool,
|
||||
{
|
||||
for i in 1..d.len() {
|
||||
let mut n = i;
|
||||
while n > 0 && is_less(&d[n], &d[n - 1]) {
|
||||
d.swap(n, n - 1);
|
||||
n -= 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn quick_sort<T, F>(d: &mut [T], is_less: &F)
|
||||
where
|
||||
F: Fn(&T, &T) -> bool,
|
||||
{
|
||||
if d.len() > 30 {
|
||||
let mut mid = partition(d, is_less);
|
||||
if mid < d.len() / 2 {
|
||||
mid += 1;
|
||||
}
|
||||
let (left, right) = d.split_at_mut(mid);
|
||||
quick_sort(left, is_less);
|
||||
quick_sort(right, is_less);
|
||||
} else {
|
||||
insert_sort(d, is_less);
|
||||
}
|
||||
}
|
||||
|
||||
fn par_quick_sort<T, F>(d: &mut [T], is_less: &F)
|
||||
where
|
||||
F: Fn(&T, &T) -> bool + Send + Sync,
|
||||
T: Send,
|
||||
{
|
||||
if d.len() > 30 {
|
||||
let mut mid = partition(d, is_less);
|
||||
if mid < d.len() / 2 {
|
||||
mid += 1;
|
||||
}
|
||||
let (left, right) = d.split_at_mut(mid);
|
||||
|
||||
if right.len() > 100_000 {
|
||||
join(
|
||||
|| par_quick_sort(left, is_less),
|
||||
|| par_quick_sort(right, is_less),
|
||||
);
|
||||
} else {
|
||||
quick_sort(left, is_less);
|
||||
quick_sort(right, is_less);
|
||||
}
|
||||
} else {
|
||||
insert_sort(d, is_less);
|
||||
}
|
||||
}
|
||||
|
||||
pub trait QSort<T> {
|
||||
fn qsort(&mut self);
|
||||
|
||||
fn is_sorted(&self) -> bool;
|
||||
}
|
||||
|
||||
impl<T> QSort<T> for [T]
|
||||
where
|
||||
T: Ord,
|
||||
{
|
||||
fn qsort(&mut self) {
|
||||
quick_sort(self, &|a: &T, b: &T| a.lt(b));
|
||||
}
|
||||
|
||||
fn is_sorted(&self) -> bool {
|
||||
self.windows(2).all(|w| w[0] <= w[1])
|
||||
}
|
||||
}
|
||||
|
||||
pub trait ParQSort<T> {
|
||||
fn par_qsort(&mut self);
|
||||
}
|
||||
|
||||
impl<T> ParQSort<T> for [T]
|
||||
where
|
||||
T: Ord + Send,
|
||||
{
|
||||
fn par_qsort(&mut self) {
|
||||
par_quick_sort(self, &|a: &T, b: &T| a.lt(b));
|
||||
}
|
||||
}
|
||||
|
||||
use rand::distributions::{Distribution, Uniform};
|
||||
use rand::thread_rng;
|
||||
use rayon::prelude::*;
|
||||
use std::time::Instant;
|
||||
use std::fs;
|
||||
|
||||
fn main() {
|
||||
let len = 50_000_000;
|
||||
let orig: Vec<i32> = Uniform::from(0..1_000_000_000)
|
||||
.sample_iter(&mut thread_rng())
|
||||
.take(len)
|
||||
.collect();
|
||||
let input = fs::read_to_string("../data/perm50e3.txt")
|
||||
.expect("Could not read file")
|
||||
.lines()
|
||||
.map(|s| s.parse::<u64>().unwrap())
|
||||
.collect::<Vec<u64>>();
|
||||
|
||||
macro_rules! test_sort {
|
||||
($name:expr, $sort:ident) => {
|
||||
println!(
|
||||
"Sorting {} million numbers with {} in Rust ...",
|
||||
len / 1_000_000,
|
||||
$name
|
||||
);
|
||||
let mut data = orig.clone();
|
||||
let start = Instant::now();
|
||||
data.$sort();
|
||||
println!("Time: {:.2?}", start.elapsed());
|
||||
assert!(data.is_sorted());
|
||||
}
|
||||
}
|
||||
|
||||
test_sort!("naive quicksort", qsort);
|
||||
test_sort!("stdlib quicksort", sort_unstable);
|
||||
test_sort!("naive parallel quicksort", par_qsort);
|
||||
test_sort!("Rayon quicksort", par_sort_unstable);
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
use super::*;
|
||||
use quickcheck::quickcheck;
|
||||
|
||||
quickcheck! {
|
||||
fn test_all(d: Vec<i32>) -> bool {
|
||||
let mut d = d;
|
||||
|
||||
let mut expected = d.clone();
|
||||
expected.sort_unstable();
|
||||
d.par_qsort();
|
||||
|
||||
expected == d
|
||||
}
|
||||
}
|
||||
println!("{}", input.len());
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user