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

/// k is 0-indexed
/// O(K)

pub fn kth_set_bit(
    mut n: u64,
    k: usize,
) -> usize {
    assert!(n.count_ones() as usize > k);

    for _ in 0..k {
        n &= n - 1;
    }

    n.trailing_zeros() as usize
}

#[cfg(test)]

mod tests {

    use super::*;

    #[test]

    fn test() {
        let n = 0b100101;

        assert_eq!(kth_set_bit(n, 0), 0);

        assert_eq!(kth_set_bit(n, 1), 2);

        assert_eq!(kth_set_bit(n, 2), 5);
    }
}