blob: e0358eeee56c50bdb67aff90ba0401ce1498cfb8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
use rand;
use rand::distributions::{IndependentSample, Range};
use actions::*;
use math::*;
use knowledge::*;
use ships::*;
pub fn shoot_smartly(knowledge: &Knowledge) -> Action {
let (weapon, target) = if knowledge.has_unknown_hits() {
destroy_shoot(&knowledge)
}
else {
seek_shoot(&knowledge)
};
Action::Shoot(weapon, target)
}
fn seek_shoot(knowledge: &Knowledge) -> (Weapon, Point) {
let (weapon, possibilities) = knowledge.get_best_seek_shots();
if possibilities.is_empty() {
println!("All possible shots on the current lattice have been tried!");
(Weapon::SingleShot, Point::new(0,0))
}
else {
let mut rng = rand::thread_rng();
let between = Range::new(0, possibilities.len());
let i = between.ind_sample(&mut rng);
(weapon, possibilities[i as usize])
}
}
fn destroy_shoot(knowledge: &Knowledge) -> (Weapon, Point) {
let possibilities = knowledge.get_best_adjacent_shots();
if possibilities.is_empty() {
seek_shoot(&knowledge)
}
else {
let mut rng = rand::thread_rng();
let between = Range::new(0, possibilities.len());
let i = between.ind_sample(&mut rng);
(Weapon::SingleShot, possibilities[i as usize])
}
}
|