Solved 0779

Author: guangsheng.li01

src/a0779_k_th_symbol_in_grammar.rs

@@ -0,0 +1,34 @@
+/*
+ * [0779] k-th-symbol-in-grammar
+ */
+
+struct Solution;
+
+impl Solution {
+    pub fn kth_grammar(n: i32, k: i32) -> i32 {
+        let mut k = k;
+        let mut n = n - 1;
+        let mut p = false;
+        while n > 0 {
+            n -= 1;
+            if k % 2 == 0 {
+                p = !p;
+            }
+            k = (k - 1) / 2 + 1;
+        }
+        p as i32
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_case0() {
+        assert_eq!(Solution::kth_grammar(1, 1), 0);
+        assert_eq!(Solution::kth_grammar(2, 1), 0);
+        assert_eq!(Solution::kth_grammar(2, 2), 1);
+        assert_eq!(Solution::kth_grammar(4, 5), 1);
+    }
+}

src/lib.rs

@@ -36,6 +36,7 @@ mod a0654_maximum_binary_tree;
 mod a0701_insert_into_a_binary_search_tree;
 mod a0726_number_of_atoms;
 mod a0761_special_binary_string;
+mod a0779_k_th_symbol_in_grammar;
 mod a0783_minimum_distance_between_bst_nodes;
 mod a0867_transpose_matrix;
 mod a0883_projection_area_of_3d_shapes;