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
29
30
31
32
33
34
35
36
37
38
39
40

use crate::dynamic_shaped_matrix::Matrix;

impl<T: Clone> Matrix<T> {
    pub fn transpose(&self) -> Self {
        let (h, w) = self.shape();

        let mut a = vec![vec![None; h]; w];

        for i in 0..h {
            for j in 0..w {
                a[j][i] = Some(self[i][j].clone());
            }
        }

        a.into_iter()
            .map(|row| row.into_iter().map(|x| x.unwrap()).collect())
            .collect::<Vec<_>>()
            .into()
    }
}

#[cfg(test)]

mod tests {

    use super::*;

    #[test]

    fn test() {
        type Mat = Matrix<usize>;

        let a: Mat = [[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]].into();

        assert_eq!(
            a.transpose(),
            [[0, 4, 8], [1, 5, 9], [2, 6, 10], [3, 7, 11]].into()
        );
    }
}