mirror of
https://gitlab.com/Anson-Projects/anson-stuff/Go-v-Rust-Quicksort.git
synced 2025-06-16 14:16:38 +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;
|
use std::fs;
|
||||||
|
|
||||||
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;
|
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
let len = 50_000_000;
|
let input = fs::read_to_string("../data/perm50e3.txt")
|
||||||
let orig: Vec<i32> = Uniform::from(0..1_000_000_000)
|
.expect("Could not read file")
|
||||||
.sample_iter(&mut thread_rng())
|
.lines()
|
||||||
.take(len)
|
.map(|s| s.parse::<u64>().unwrap())
|
||||||
.collect();
|
.collect::<Vec<u64>>();
|
||||||
|
|
||||||
macro_rules! test_sort {
|
println!("{}", input.len());
|
||||||
($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
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
Loading…
x
Reference in New Issue
Block a user