blob: 4e621a31a43123900dd779e91baa089b18ad5777 (
plain)
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
#!/usr/bin/ruby -w
# ch-1.rb
# > https://theweeklychallenge.org/blog/perl-weekly-challenge-146/
#
# ## Task 1 > 10001st Prime Number
# ================================
#
# Write a script to generate the 10001st prime number.
################################################################################
# PWC Solution #################################################################
################################################################################
def get_prime(nth)
primes = [2, 3]
cursor = 5
until primes.length >= nth
is_prime = true
primes.each do |prime|
(prime > Math.sqrt(cursor) || !is_prime) && break
(cursor % prime).zero? && is_prime = false
end
is_prime && primes.push(cursor)
cursor += 2
end
nth <= 2 ? primes[nth - 1] : primes.pop
end
################################################################################
# Utilities ####################################################################
################################################################################
def get_suffix(num)
last = num.to_s[-1].to_i
return 'th' if last.zero? || last >= 4
return 'st' if last == 1
return 'nd' if last == 2
return 'rd' if last == 3
''
end
def print_colors(str, color)
reset = "\e[0m"
case color
when 'yellow'
"\e[33m#{str}#{reset}"
when 'green'
"\e[32m#{str}#{reset}"
else
str
end
end
################################################################################
# Main #########################################################################
################################################################################
nth = ARGV[0] ? ARGV[0].to_i : 10_001
prime = get_prime(nth)
suffix = get_suffix(nth)
num_str = print_colors(nth.to_s + suffix, 'yellow')
prime_str = print_colors(prime, 'green')
puts "The #{num_str} prime number is #{prime_str}"
|
