dsalgo 0.3.10

A package for Datastructures and Algorithms.
Documentation 
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

pub fn argmax<T: PartialOrd + Clone>(a: &[T]) -> usize {
    let (mut idx, mut mx) = (0, &a[0]);

    for (i, x) in a.iter().enumerate() {
        if x > mx {
            idx = i;

            mx = x;
        }
    }

    idx
}

#[cfg(test)]

mod tests {

    use super::*;

    #[test]

    fn test() {
        let a = vec![3, 4, 0, 5, 0];

        assert_eq!(argmax(&a), 3);
    }
}