Решение на Домашно 1 - търсене на съкровища от Емилиан Янев
Резултати
- 16 точки от тестове
- 0 бонус точки
- 16 точки общо
- 4 успешни тест(а)
- 1 неуспешни тест(а)
Код
use std::sync::mpsc;
use std::collections::HashMap;
use std::sync::{
atomic::Ordering,
Arc,
atomic::AtomicBool,
};
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct TreasureLoc {
pub lane_index: usize,
pub cell_coord: usize,
pub value: i32,
}
#[derive(Debug, PartialEq, Eq)]
pub enum FoundTreasures {
Big(TreasureLoc),
Small(Vec<TreasureLoc>),
Nothing,
}
enum Message {
Treasure(TreasureLoc),
Done(usize)
}
pub struct Scan<'a> {
pub start_coord: usize,
pub cells: &'a [i32],
}
pub struct Drone {
lane_index: usize,
send_channel: mpsc::Sender<Message>,
stop: Arc<AtomicBool>
}
impl Drone {
pub fn explore(&mut self, scanner: &mut dyn Iterator<Item = Scan<'_>>) {
for scan in scanner {
if self.stop.load(Ordering::Relaxed) {
break;
}
for (i, &v) in scan.cells.iter().enumerate() {
if self.stop.load(Ordering::Relaxed) {
break;
}
if v > 0 {
let loc = TreasureLoc {
lane_index: self.lane_index,
cell_coord: scan.start_coord + i,
value: v,
};
if self.send_channel.send(Message::Treasure(loc)).is_err() {
self.stop.store(true, Ordering::Relaxed);
return;
}
}
}
}
let _ = self.send_channel.send(Message::Done(self.lane_index));
}
}
pub struct DroneController {
stop: Arc<AtomicBool>,
send_channel: mpsc::Sender<Message>,
receive_channel: mpsc::Receiver<Message>,
drones_count: usize
}
impl DroneController {
pub fn new() -> Self {
let(send_channel, receive_channel) = mpsc::channel();
let stop = Arc::new(AtomicBool::new(false));
let drones_count = 0;
Self {
stop,
send_channel,
receive_channel,
drones_count
}
}
pub fn create_drone(&mut self, lane_index: usize) -> Drone {
let drone_send_channel = self.send_channel.clone();
self.drones_count += 1;
Drone{lane_index,
send_channel: drone_send_channel,
stop: self.stop.clone()
}
}
pub fn run(&mut self) -> FoundTreasures {
let mut small_treasure: i32 = 0;
let mut idx = 0;
let mut small_treasures: HashMap<usize, TreasureLoc> = HashMap::new();
while let Ok(message) = self.receive_channel.recv() {
match message {
Message::Treasure(loc) => {
if loc.value >= 999 {
self.stop.store(true, Ordering::Relaxed);
return FoundTreasures::Big(loc);
}
if !small_treasures.contains_key(&loc.lane_index) {
small_treasure += loc.value;
small_treasures.insert(loc.lane_index, loc);
if small_treasure >= 300 {
self.stop.store(true, Ordering::Relaxed);
let mut v: Vec<TreasureLoc> =
small_treasures.into_values().collect();
v.sort_by_key(|t| t.lane_index);
return FoundTreasures::Small(v);
}
}
}
Message::Done(_lane) => {
idx += 1;
if idx == self.drones_count {
break;
}
}
}
}
FoundTreasures::Nothing
}
}
Лог от изпълнението
Updating crates.io index
Locking 46 packages to latest compatible versions
Compiling proc-macro2 v1.0.104
Compiling unicode-ident v1.0.22
Compiling quote v1.0.42
Compiling libc v0.2.178
Compiling syn v2.0.111
Compiling futures-core v0.3.31
Compiling pin-project-lite v0.2.16
Compiling futures-sink v0.3.31
Compiling parking_lot_core v0.9.12
Compiling futures-channel v0.3.31
Compiling futures-io v0.3.31
Compiling slab v0.4.11
Compiling memchr v2.7.6
Compiling futures-task v0.3.31
Compiling pin-utils v0.1.0
Compiling scopeguard v1.2.0
Compiling smallvec v1.15.1
Compiling cfg-if v1.0.4
Compiling lock_api v0.4.14
Compiling errno v0.3.14
Compiling signal-hook-registry v1.4.8
Compiling parking_lot v0.12.5
Compiling socket2 v0.6.1
Compiling futures-macro v0.3.31
Compiling tokio-macros v2.6.0
Compiling mio v1.1.1
Compiling futures-util v0.3.31
Compiling bytes v1.11.0
Compiling tokio v1.48.0
Compiling futures-executor v0.3.31
Compiling futures v0.3.31
Compiling solution v0.1.0 (/tmp/d20251229-4108951-1fne9t8/solution)
Finished `test` profile [unoptimized + debuginfo] target(s) in 17.84s
Running tests/solution_test.rs (target/debug/deps/solution_test-f512224d9fb3caf8)
running 5 tests
test solution_test::test_nothing ... ok
test solution_test::test_big_treasure ... ok
test solution_test::test_small_treasure ... ok
test solution_test::test_small_treasure_2 ... FAILED
test solution_test::test_return_immediately_when_found ... ok
failures:
---- solution_test::test_small_treasure_2 stdout ----
thread 'solution_test::test_small_treasure_2' panicked at tests/solution_test.rs:144:14:
assertion failed
expected: FoundTreasures::Small(_)
found: Nothing
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
failures:
solution_test::test_small_treasure_2
test result: FAILED. 4 passed; 1 failed; 0 ignored; 0 measured; 0 filtered out; finished in 0.10s
error: test failed, to rerun pass `--test solution_test`