ID

67c7977

date

2025-01-09 23:45:24+00:00

author

Alex Chan <alex@alexwlchan.net>

parent

461d4f6

message

Formatting; we don't need to print `k`

changed files

2 files, 7 additions, 21 deletions

• src/main.rs
• src/sampling.rs

Changed files

src/main.rs (1402) → src/main.rs (1075)

diff --git a/src/main.rs b/src/main.rs
index 260dd60..5389f36 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -29,31 +29,17 @@ fn main() {
         }
     };
 
-    let lines = std::io::stdin().lock().lines()
-        .map(|line| match line {
-          Ok(ln) => ln,
-          Err(e) => {
+    let lines = std::io::stdin().lock().lines().map(|line| match line {
+        Ok(ln) => ln,
+        Err(e) => {
             eprintln!("Unable to read from stdin: {:?}", e);
             std::process::exit(1)
-          }
-        });
-
-    println!("k = {:?}", k);
+        }
+    });
 
     let sample = sampling::reservoir_sample(lines, k);
 
     for line in sample {
-      println!("{}", line);
+        println!("{}", line);
     }
-
-
-        // Read the first N lines from stdout
-        // let stdin = io::stdin();
-
-        // let a = [1, 2, 3, 4, 5, 6];
-        // let iter = a.iter();
-        //
-        // println!("{:?}", sampling::reservoir_sample(iter, n));
-        //
-        // println!("n = {:?}", n);
 }

src/sampling.rs (5970) → src/sampling.rs (5972)

diff --git a/src/sampling.rs b/src/sampling.rs
index 06a3d04..ba679d4 100644
--- a/src/sampling.rs
+++ b/src/sampling.rs
@@ -17,7 +17,7 @@ pub fn reservoir_sample<T: std::fmt::Debug>(
     // Taking a sample with k=0 doesn't make much sense in practice,
     // but we include this to avoid problems downstream.
     if k == 0 {
-      return vec![];
+        return vec![];
     }
 
     // Create an empty reservoir.