From 0b2cbd28e2f47ac7d0831a139b9543cb19ba936f Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Tue, 30 Aug 2016 04:05:06 +0200 Subject: Solve Nth prime --- go/nth-prime/nth_prime.go | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 go/nth-prime/nth_prime.go (limited to 'go/nth-prime/nth_prime.go') diff --git a/go/nth-prime/nth_prime.go b/go/nth-prime/nth_prime.go new file mode 100644 index 0000000..7e01d8e --- /dev/null +++ b/go/nth-prime/nth_prime.go @@ -0,0 +1,42 @@ +package prime + +func gen() (chan int, chan bool) { + stop := make(chan bool) + ch := make(chan int) + go func() { + defer close(ch) + for i := 2; ; i++ { + select { + case ch <- i: + case <-stop: + return + } + } + }() + return ch, stop +} + +func filter(in chan int, prime int) chan int { + ch := make(chan int) + go func() { + defer close(ch) + for i := range in { + if i%prime != 0 { + ch <- i + } + } + }() + return ch +} + +func Nth(n int) (int, bool) { + if n < 1 { + return 0, false + } + ch, stop := gen() + defer func() { stop <- true }() + for i := 1; i < n; i++ { + ch = filter(ch, <-ch) + } + return <-ch, true +} -- cgit v1.2.3