From 4b0a6a01b051a4ebfbc17661d14cb23fe4f275fb Mon Sep 17 00:00:00 2001 From: Orangerot Date: Thu, 27 Jun 2024 11:30:16 +0200 Subject: Initial commit --- k-th-symbol-in-grammar/src/main.rs | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 k-th-symbol-in-grammar/src/main.rs (limited to 'k-th-symbol-in-grammar/src') diff --git a/k-th-symbol-in-grammar/src/main.rs b/k-th-symbol-in-grammar/src/main.rs new file mode 100644 index 0000000..5aaa24c --- /dev/null +++ b/k-th-symbol-in-grammar/src/main.rs @@ -0,0 +1,28 @@ +fn main() { + println!("Hello, world!"); + let tests = vec![ + (1,1,0), + (2,1,0), + (2,2,1), + (30, 434991989, 0) + ]; + for (n,k,o) in tests { + let result = Solution::kth_grammar(n,k); + println!("{n} {k} {result} {o}"); + } +} + +struct Solution; + +impl Solution { + pub fn kth_grammar(n: i32, k: i32) -> i32 { + let mut s: Vec = vec![false]; + for _ in 0..n { + s = s.iter().flat_map(|x| match x { + false => [false,true], + true => [true,false] + }).collect::>(); + } + s[(k - 1) as usize] as i32 + } +} -- cgit v1.2.3